pytorch安装记录

1.pycharm和anaconda按照别人的教程进行安装
教程1
2.cuda和cudnn按照教程安装
教程2
注意:显卡因为是RTX3050Ti,对应下载的cuda版本必须要是在11.之上,还有windos版本问题,不然会显示不匹配问题。
3.pytorch安装
3.1在最近添加和anaconda包中找到anaconda prompt
在这里插入图片描述
3.2出现这种说明成功
pytorch安装记录_第1张图片
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
pytorch安装记录_第2张图片
只复制框框里面的语句,后面的-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,就说明安装正确了。

你可能感兴趣的:(pytorch,python,深度学习)