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