【PYTORCH,TENSORFLOW环境配置,安装,自用代码】

conda -V(查看版本,这步不要也罢)

conda create -n test python=3.7(创建环境)

conda activate test(激活)

conda env list(查看自己的环境)

nvidia-smi(查看是否可以安装GPU版本)

PyTorch(看自己能安装什么版本)

复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121(本人用的)

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

(之前的,加了清华大学的镜像源)

Tensorflow安装

conda search -f cudatoolkit

conda search -c conda-forge cudatoolkit

conda install cudatoolkit==11.0.3 -c conda-forge

conda search -c conda-forge cudnn

conda install cudnn=8.0.5.39 -c conda-forge(本人用的,还可以用更新的)

在 Windows 环境中从源代码构建 | TensorFlow (google.cn)

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.4.0

(真麻烦啊)

PS:用终端安装才是真快

!!!!!安装真的好麻烦,好麻烦!!!!!!!!!!!!!!!!!!!!!!!!!!谁懂,并不是为了省淘宝安装费,而是配置环境是最基础的!!!!!!(不得不学会)

(按照顺序无脑打得了,已经写的足够详细了,腰疼,配置了好几天,因为一些很愚蠢的问题,所以这些是精华)

相关推荐
宿州派大星3 分钟前
[NLP实战] 基于PyTorch实现N-gram词嵌入模型:输入4个词预测第5个词
人工智能·pytorch·深度学习·nlp
kuuailetianzi12 分钟前
Python初识:定位、优势与发展历程
python
guoran_shini22 分钟前
LLM微调-训练垂类问答模型
python·lora·sft
qq_1998868728 分钟前
第8板块·第2节:统一内存的高级特性与性能调优
c++·人工智能·gpu算力·cuda
basketball61628 分钟前
Python FastAPI 介绍以及常用方法
python·fastapi·vllm·ai infra
ting945200030 分钟前
深度拆解Enter Pro AI原生开发平台:从底层架构到企业级落地技术实践
人工智能·架构·ai-native
lisw0532 分钟前
人工智能辅助科学的快与慢!
人工智能
9呀38 分钟前
vscode如何打开多个codex标签页
人工智能
todoitbo42 分钟前
本地图库语义搜索实战:接上蓝耘元生代,让“傍晚的海边“能搜到图
人工智能·ai·api·工具实战
论文复现现场1 小时前
企业知识库 RAG 用什么模型便宜?GLM-5.3-Flash 长文本 API 实测
python·rag·企业知识库·大模型api·glm-5.3-flash