Windows10+Anaconda+pytorch(GPU–CUDA 9.0)+Pycharm(避坑)

第一步 Anaconda安装

进入Anaconda官网

pytorch cudatookit对应版本 pytorch cuda9.0_python


找到合适的版本安装,在安装过程中记得要勾选环境变量的选型。第二步 查看本机CUDA版本

1.打开控制面板

2.在右上角查看方式选择小图标

pytorch cudatookit对应版本 pytorch cuda9.0_pytorch_02


进入NVIDIA 控制面板

3.点击帮助,系统信息进入系统信息界面,选择组件

pytorch cudatookit对应版本 pytorch cuda9.0_pytorch_03


4.NVCUDA.DLL上显示的是CUDA当前版本号(我这里是9.0)

pytorch cudatookit对应版本 pytorch cuda9.0_深度学习_04


第三步 CUDA9.0+CUDNN7.05安装

1.CUDA Toolkit 9.0 Downloads

pytorch cudatookit对应版本 pytorch cuda9.0_python_05


找到本机系统安装(其他版本进入CUDA Toolkit 官网安装)2.安装注意:安装cuda时第一次设置的是临时解压目录,第二次设置的才是安装路径,临时解压目录可以自定义,但个人认为没必要,因为cuda安装成功之后会自行删除。

在CUDA安装时有一步是自定义安装选型,选择驱动组件的,勾选组件时根据个人需求安装,CUDA这个选型建议全勾选。

pytorch cudatookit对应版本 pytorch cuda9.0_pytorch_06


在安装组件的过程中360安全卫士会弹出阻止安装框,允许即可。或者在安装时关闭一会儿360安全卫士。

3.安装完成后,先查看系统变量,然后需要添加cuda的系统变量

安装完成之后就有的:

pytorch cudatookit对应版本 pytorch cuda9.0_pytorch_07


需要添加的:

pytorch cudatookit对应版本 pytorch cuda9.0_pycharm_08


Path中需要添加的:

pytorch cudatookit对应版本 pytorch cuda9.0_cuda_09


4.cudnn下载 cuDNN官网 注意这里cudnn的版本需要和CUDA一致我这里下载的版本如下图所示

pytorch cudatookit对应版本 pytorch cuda9.0_cuda_10


解压之后,将解压之后的文件夹下的bin、include、lib文件拷贝到cuda安装目录对应的文件下。检查是否安装成功:

WIN+R输入cmd,cd到安装目录的…\extras\demo_suite文件夹下:

分别执行bandwidthTest.exe和deviceQuery.exe,如果均出现PASS,则说明安装成功。

pytorch cudatookit对应版本 pytorch cuda9.0_pycharm_11


pytorch cudatookit对应版本 pytorch cuda9.0_pytorch_12

第四步、pytorch安装

这里强烈推荐用离线下载的方式安装,具体操作如下:

1.进入pytorch官方网站,查找执行命令

pytorch cudatookit对应版本 pytorch cuda9.0_python_13


上面没有CUDA9.0版本,因此进入previous-versions网站查找以前版本的安装命令,个人强烈推荐使用pip命令安装pytorch.

2.可以在线下载也可以离线下载,个人推荐离线下载

由于pip install极易出现超时提醒,因此可以加清华源、阿里源,科大的源等,这个随意。

pip时加上参数 -i https://pypi.tuna.tsinghua.edu.cn/simple(清华源)

或者 -i http://mirrors.aliyun.com/pypi/simple/ (阿里源)

或者 -i https://pypi.mirrors.ustc.edu.cn/simple/ (中科大源)

方法一:打开anaconda prompt:三条语句
1.python -m pip install --upgrade pip
2.pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl -i http://mirrors.aliyun.com/pypi/simple/
3.pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl -i http://mirrors.aliyun.com/pypi/simple/

方法二:
1.先下载
https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl

https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl 这两个

2.打开anaconda promt:

3.进入.whl所在的磁盘; cd 切到所在路径 ; pip install + 文件名字

pytorch cudatookit对应版本 pytorch cuda9.0_cuda_14


测试:在anaconda promt输入python

pytorch cudatookit对应版本 pytorch cuda9.0_深度学习_15


第五步、Pycharm安装

Pycharm官网下载本机版本,有个人版和社区版,个人版是免费的。

pytorch cudatookit对应版本 pytorch cuda9.0_深度学习_16


PyCharm安装完成之后需要解释器,创建一个新的项目,输入项目工作区、项目名,仍需注意的是解释器的选择。

pytorch cudatookit对应版本 pytorch cuda9.0_深度学习_17


pytorch cudatookit对应版本 pytorch cuda9.0_深度学习_18


选择Anaconda中python.exe所在路径即可。

最后,创建一个.py,运行一下,没有报错证明工程创建正确。

pytorch cudatookit对应版本 pytorch cuda9.0_pycharm_19


pytorch cudatookit对应版本 pytorch cuda9.0_深度学习_20


当然也可以搜一个需要使用GPU的代码,运行看是否运行成功。

到此,环境全部搭建结束了!!!