安装pytorch过程
-
首先安装anaconda
-
创建环境
-
pytorch安装
-
检验安装
1.安装anaconda(我的电脑已经安装过anaconda,所以这里跳过)
如果没有安装,可以参考 Anaconda超详细安装教程(Windows环境下)_conda安装-CSDN博客
2.创建环境(pytorch)
首先打开Anaconda Prompt
然后创建环境(环境名为pytorch,当前的环境python=3.9.7)
是否继续(y 继续)
验证环境是否创建成功,输入activate pytorch(pytorch是你定义的这个环境的名字),左边 的环境就从base(基本环境),变成了pytorch环境
3.pytorch安装
查看CUDA版本数,进入命令符号窗口,输入nvidia-smi,查看当前驱动的版本号,观察 Driver Version的值是否大于400,如果小于请更新显卡驱动
进入pytorch官网 PyTorch
选择适应的版本(查看自己的CUDA版本数,我的cuda是10.2)
复制这段指令(由于从官网下载过慢,容易导致失败,我换成了清华源)
pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
4.检验安装
先输入python,然后输入import torch,如果输入后没有任何报错,然后再输入 torch.cuda.is_available(),返回的是True,就成功了