win10中pytorch以及torchvision安装填坑

安装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:

win10中pytorch以及torchvision安装填坑_第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

win10中pytorch以及torchvision安装填坑_第2张图片

显示所有版本适应,并且程序能正确运行

 

 

 

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