有20天没更新了,8.3号从成都回到山东,在家待了几天然后去烟台找对象过节了,正值秋招季节,各种忙碌,偷得片刻闲暇,整理一篇帮对象装Pytorch时遇到的问题,都是很基础的东西,但真正遇到了却也不是那么容易解决,那就开始吧~
首先系统是win10专业版,为了方便安全我们使用Anaconda来配置环境,关于Anaconda的安装就不再表述了,网上教程一大堆,也的确没有什么值得注意的,唯一的一点就是选择是否将路径添加到环境变量这一选项,很多博主都说不能选这一项,后续自己动手配置就ok,我一开始是选了的,后来因为碰到一些问题就给删除了,自己动手添加的(如果你选择自动添加路径,他会在环境变量中增加四条或五条信息,动手添加的话只需两条信息即可,关于这一点大家可以自行查阅)
安装Anaconda后我们创建虚拟环境来管理Pytorch,这时报错HTTPEoor,然后我们可以使用快捷键“win+R”打开运行界面,输入cmd打开命令窗口,在其中依次输入以下几条命令,切记不可在anaconda prompt中输入,这是一个大坑!!!
# 添加Anaconda的清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 设置搜索时显示通道地址
config --set show_channel_urls yes
虚拟环境创建完成后就可以在其中自由飞翔了,由于之前已经装好了CUDA,那就可以直接安装对应版本的Pytorch了,这里的版本问题一定要注意,这是很多后续问题的根源,我们可以打开英伟达控制面板查看CUDA的版本,显示为9.2.148:
也可以在命令窗口中输入nvcc -V来查看:
按照一般操作,我们可以到pytorch官网去看一看:https://pytorch.org/get-started/locally/
主界面如下所示:
这种形式很人性化,直接选择就行了,一开始选的是conda,结果又报错了:
CondaHTTPError: HTTP 000 CONNECTION FAILED for url
然后选用pip,由于没有安装pip3,所以用的pip,官网安装很慢,去掉上方Run this Command中后边的官网链接也是报错,无法用前边设置的清华镜像安装(也可能是我前边设置有问题,才导致无法使用清华镜像),于是在浏览器中下载whl文件然后选择相应的版本下载,最后本地安装
具体操作是在此地址下载对应的pytorch版本:https://download.pytorch.org/whl/torch_stable.html
在本例中找到对照项:
下载完毕后进入Anaconda的虚拟环境(本例虚拟环境名为pytorch),使用匹配安装:
安装成功后继续安装该虚拟环境下的spyder,建议在Anaconda Navigator中直接点击安装,完成后在Anaconda下会有两个spyder,分别是属于base和pytorch环境的,我们可以测试一下代码:
import torch
flag = torch.cuda.is_available()
print(flag)
ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())
看下效果:
如果大家在安装过程中遇到什么问题可以在下方留言,再会~~
关于torchvision版本问题请大家参照此博客:https://blog.csdn.net/yayadazui/article/details/99738114
公众号为“非著名IT表演艺术家”,比较中二的名字,就是灵光一闪,然后这个名字就冒出来了……