pytorch：Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1

一、安装python3.6

1.1 下载

https://www.python.org/downloads/ 下载python3.6或其他版本

1.2 安装并验证是否可用

借鉴：https://blog.csdn.net/hpu_yly_bj/article/details/78918526

二、安装PyCharm

借鉴：https://www.94e.cn/info/5451

三、配置测试CUDA9.0和cudnn7

3.1 安装CUDA9.0

https://developer.nvidia.com/cuda-toolkit-archive 选择合适的版本

安装完成后系统变量会添加较多路径

3.2、下载对应的cudnn

https://developer.nvidia.com/rdp/cudnn-download

将解压后的三个文件夹（bin、lib、include）复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

3.3 测试是否安装成功

Win+R

cmd

nvcc- V

四、安装pytorch

Pytorch官网 https://pytorch.org/

之前的版本在 https://pytorch.org/get-started/previous-versions/

 

在PyCharm的Terminal中分别输入

pip3 install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl

Successfully …

 

pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

Successfully …

 

成功后测试一下就OK了