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

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

第一步 Anaconda安装
进入Anaconda官网

找到合适的版本安装，在安装过程中记得要勾选环境变量的选型。

第二步 查看本机CUDA版本
1.打开控制面板
2.在右上角查看方式选择小图标

进入NVIDIA 控制面板
3.点击帮助，系统信息进入系统信息界面，选择组件

4.NVCUDA.DLL上显示的是CUDA当前版本号(我这里是9.0）
第三步 CUDA9.0+CUDNN7.05安装

1.CUDA Toolkit 9.0 Downloads

找到本机系统安装（其他版本进入CUDA Toolkit 官网安装）

2.安装注意：安装cuda时第一次设置的是临时解压目录，第二次设置的才是安装路径，临时解压目录可以自定义，但个人认为没必要，因为cuda安装成功之后会自行删除。
在CUDA安装时有一步是自定义安装选型，选择驱动组件的，勾选组件时根据个人需求安装，CUDA这个选型建议全勾选。

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

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

安装完成之后就有的：

需要添加的：

Path中需要添加的：

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

解压之后，将解压之后的文件夹下的bin、include、lib文件拷贝到cuda安装目录对应的文件下。

检查是否安装成功：
WIN+R输入cmd，cd到安装目录的…\extras\demo_suite文件夹下：
分别执行bandwidthTest.exe和deviceQuery.exe，如果均出现PASS，则说明安装成功。

第四步、pytorch安装
这里强烈推荐用离线下载的方式安装，具体操作如下：
1.进入pytorch官方网站，查找执行命令

上面没有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 + 文件名字

测试：在anaconda promt输入python

第五步、Pycharm安装
在Pycharm官网下载本机版本,有个人版和社区版，个人版是免费的。

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


选择Anaconda中python.exe所在路径即可。
最后，创建一个.py，运行一下，没有报错证明工程创建正确。


当然也可以搜一个需要使用GPU的代码，运行看是否运行成功。
到此，环境全部搭建结束了！！！