安装pytorch以及torchvision过程中遇到了一些问题,记录下来以便日后复习,也希望给遇到相同问题的小伙伴一些启发。
1.安装pytorch
anaconda和cuda的安装过程这里不再说明,本文使用的是python3.6和cuda8.0
参考网上一些大佬的教程用conda命令执行官网上的相关命令时,安装过程实在是太慢了,使用了清华源也不能很好地得到解决。于是转战pip3;使用pytorch中文网(https://www.pytorchtutorial.com/pytorch-installation-commands/#Python_36-2)上的相关命令即可,这里我安装的是0.4.1:
2.安装torchvision
当然先只运行上方的命令,不要安装torchvision,因为会有以下报错,这里显示torch为1.1.0可能是之前conda安装造成的(未下载完就结束了)
之后查阅资料发现了以下命令,安装成功:
pip install --no-deps torchvision
以为此时大功告成,怎想运行程序时出现了报错:
from torchvision._C import * 找不到相关DLL
这应该是torchvision0.3.0和当前版本pytorch0.4.1不对应的原因
这里决定更改torchvision版本命令:
pip install torchvision==0.1.8
显示所有版本适应,并且程序能正确运行