在win下运行,会找不到pytorch和torchvision(如图所示):python
安装pytorch的方法,是在这个知乎专栏里找到的,附上地址,以示对做者的尊重:windows
https://zhuanlan.zhihu.com/p/26871672spa
安装的命令有四个,第一个表示没有cuda使用pytorch,也就是只有cpuget
其余是安装了对应版本的cuda以后,安装pytorch 的命令(忽略掉注释里的windows版本,只须要看cuda对应的版本就好~,个人cuda版本是9.1.85,最新版本,选择的是第三个命令,安装可用,都是属于cuda9),在安装的时候,最好先墙出去,速度会快些,固然不墙也不要紧,就是等的久一点而已嘛~io
这样,pytorch就安装完毕了,可是还有pytorchvision须要安装,在使用时,pytorch和pytorchvision每每须要同时出现,pytorchvision安装方法,我是经过本地安装的方式进行安装的,本地包的下载地址为:https://pypi.python.org/pypi/torchvision/0.1.8)
下载
下载好以后,在当前文件夹下打开命令窗,输入下面的命令,安装pytorchvision:方法
至此,全部的安装就已经完成。im
下面查看是否安装好pytorch,torchvision和pytorch是否支持cuda,输出为True,证实ok。专栏