win10+cuda9.1+cudnn-9.1-windows10-x64-v7.1.3+pytorch-0.4.1-py37_cuda90_cudnn7he774522_1
新手第一次安装,竟然安装了一天,醉了醉了。
写在最前面,安装第三步安装pytorch和torchvision时,最后是用conda新建一个虚拟环境,防止破坏原始的python环境条件,用conda创建一个虚拟环境对于以后做项目来讲,不需要有无用的包造成影响等。
1、上一篇文章说了,先看自己电脑对应的cuda版本型号,这里就不多说了。
直接上连接点击就直接跳转cuda。找到自己需要的版本直接下载就行。
安装的时候默认傻瓜式操作就可以,中间值勾选第一个就可以,其余的不需要。
2、第二步就是下载cudnn了,我的电脑是对应的7.1.3,当然只要他对应的是cuda是9.1版本就行,但是官网更新之后就没有9.1版本的了,可以选择对应版本是9.0的,向下兼容。跳转连接cudnn这个需要先注册,注册之后才能下载。
3.完成上面两部分基本上就完成三分之一了。安装pytorch的难处在于不知道版本匹配,得自己去弄明白,我也是瞎碰上了。
pytorch我安装的是pytorch-0.4.1gpu版本的,我一开始安装的是pytorch-1.1.0版本的,但是这种不知道为什么老是显示找不到模块,看了好多博客怎么改,都没有改好,结果换了一个版本,就行了。
pytorch的安装,我是用的conda安装,这样的话就不需要下载whl文件,