环境为win8.1+python3.6.2
这里选择通过Python3.6+pip安装gpu版本pytorch
首先要先安装cuda:cuda下载地址
根据自己的系统选择相对应的版本:
下载cuDNN库:cuDNN库下载地址
需要注册并且填写一个问卷
然鹅!!!
好不容易填写完问卷之后,我发现cuDNN下载只提供了win7和win10的_(:3J∠)_
然后win8的我放弃了这一步
安装完毕后测试安装结果
打开命令提示符,输入:nvcc -V
结果如下:
接下来安装pytorch:
下载:pytorch
由于我的环境为:win8.1+python3.6.2+cuda8.0
所以选择下载:
这里可供选择的两个whl文件都是要求64位Python的,由于我之前安装的Python3.6是32位的还卸载了重新安装了64位_(:3J∠)_
下载后通过pip安装whl文件
安装完成后测试: