Home

pytorch安装

2018/08/07

环境为win8.1+python3.6.2
这里选择通过Python3.6+pip安装gpu版本pytorch

首先要先安装cuda:cuda下载地址

根据自己的系统选择相对应的版本:
image

下载cuDNN库:cuDNN库下载地址
需要注册并且填写一个问卷
image
然鹅!!!
好不容易填写完问卷之后,我发现cuDNN下载只提供了win7和win10的_(:3J∠)_
image
然后win8的我放弃了这一步

安装完毕后测试安装结果
打开命令提示符,输入:nvcc -V
结果如下:
image

接下来安装pytorch:
下载:pytorch
由于我的环境为:win8.1+python3.6.2+cuda8.0
所以选择下载:
image
这里可供选择的两个whl文件都是要求64位Python的,由于我之前安装的Python3.6是32位的还卸载了重新安装了64位_(:3J∠)_

下载后通过pip安装whl文件
安装完成后测试:
image