一、安装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了