pytorch安装记录

Posted 2022-06-27 一二三四哒

1.pycharm和anaconda按照别人的教程进行安装
教程1
2.cuda和cudnn按照教程安装
教程2
注意：显卡因为是RTX3050Ti，对应下载的cuda版本必须要是在11.之上，还有windos版本问题，不然会显示不匹配问题。
3.pytorch安装
3.1在最近添加和anaconda包中找到anaconda prompt

3.2出现这种说明成功

3.3然后在此界面上输入清华镜像，为了加快下载速度
参考：教程3和 教程4
最后自己成功安装。
在初始base就添加下面镜像语句：
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes
3.4然后创建一个附加的虚拟环境
conda create -n pytorch python==3.9
3.5激活新建的虚拟环境
conda activate pytorch
3.6根据自己下载的cuda版本对应下载pytorch版本
Pytorch官网：https://pytorch.org

只复制框框里面的语句，后面的-c pytorch不要，不然镜像就没有用了，当然不使用清华镜像的话也是可以的，就把语句全复制，这样下载速度会慢
conda install pytorch torchvision torchaudio cudatoolkit=11.3
然后下载完成，此时需要检查自己下载的pytorch是cuda版本还是cpu版本，通过输入conda list

如果出现上述cpu版本，就会报错，此时就需要输入：
conda uninstall cpuonly
此语句结束后会发现pytorch变成了gpu版本

如果发现自己没有cpuonly，就采用
先：conda install cpuonly
再：conda uninstall cpuonly
此时输入：python，进入python界面
再输入：import torch
torch.cuda.is_available()
返回：True，就说明安装正确了。