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

一、安装python3.6

二、安装PyCharm

三、配置测试CUDA9.0cudnn7

3.1 安装CUDA9.0

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

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

pytorch:Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1_第1张图片

pytorch:Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1_第2张图片

pytorch:Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1_第3张图片

3.2、下载对应的cudnn

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

pytorch:Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1_第4张图片

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

3.3 测试是否安装成功

Win+R

cmd

nvcc-V

pytorch:Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1_第5张图片

四、安装pytorch

Pytorch官网 https://pytorch.org/

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

pytorch:Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1_第6张图片

 

在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了

pytorch:Win10 + python3.6 + pycharm + CUDA 9.0 安装配置pytorch 1.1_第7张图片

你可能感兴趣的:(pytorch)