首先分享一个博客的链接,我开始是在这上面学习的,之后遇见的一些问题,就自己找方法解决的。
还有一个链接也是安装教程 链接
我是自己想在windows10上面装pytorch,因为自己在虚拟机上装了ubuntu16.04,配好了tensorflow的环境。
我自己是先装好了python3.6并且电脑带有GPU的 是 grforce 940MX
查看自己是否安装在cmd nvcc -V
首先要安装显卡驱动cuda 建议安装cuda9.1版本,开始在官网找不到这个安装包不过还是找到了现在最新是cuda10.0版本
附上链接吧下载地址
下载之后就可以开始安装了
- 首先点开后,选择自定义安装,会出现安装路径,可以自己设置,但是不重要安装后自己就会消失。
2.安装时可以看到,cuda中的Visual Studio Integration不要勾选,其他的勾上,默认安装路径,然后开始安装。
3.环境变量的配置
4.查看安装是否成功
cmd后 set cuda
5.进入cuda的安装路径
以上没问题的话,就是cuda9.1安装成功了。
还要加上cudnn
下载地址官网
接下来安装pytorch吧
选择pip安装
在cmd输入
开始安装,这里也是需要翻墙的速度快,不然很慢。
之后安装torchvision,通过本地安装
下载地址 地址
让后在下载后的文件夹命令安装
最后看torch是否可以使用
在这里遇到一个问题,import torch失败
最后再这篇文章找到了,解决方法
主要是环境变量配好,配到路径中去。