Windows10+Anaconda+pytorch(GPU--CUDA 9.0)+Pycharm(避坑)
**第一步 Anaconda安装**
进入[Anaconda官网](https://www.anaconda.com/products/individual)
找到合适的版本安装,在安装过程中记得要勾选环境变量的选型。
**第二步 查看本机CUDA版本**
1.打开控制面板
2.在右上角查看方式选择小图标
进入NVIDIA 控制面板
3.点击帮助,系统信息进入系统信息界面,选择组件
4.NVCUDA.DLL上显示的是CUDA当前版本号(我这里是9.0)
**第三步 CUDA9.0+CUDNN7.05安装**
1.[CUDA Toolkit 9.0 Downloads](https://developer.nvidia.com/cuda-90-download-archive)
找到本机系统安装(其他版本进入[CUDA Toolkit 官网](https://developer.nvidia.com/cuda-toolkit-archive)安装)
2.安装*注意*:安装cuda时第一次设置的是临时解压目录,第二次设置的才是安装路径,临时解压目录可以自定义,但个人认为没必要,因为cuda安装成功之后会自行删除。
在CUDA安装时有一步是自定义安装选型,选择驱动组件的,勾选组件时根据个人需求安装,CUDA这个选型建议全勾选。
在安装组件的过程中360安全卫士会弹出阻止安装框,允许即可。或者在安装时关闭一会儿360安全卫士。
3.*安装完成后*,先查看系统变量,然后需要添加cuda的系统变量
4.cudnn下载 [cuDNN官网](https://developer.nvidia.com/rdp/cudnn-archive)
注意这里cudnn的版本需要和CUDA一致我这里下载的版本如下图所示
解压之后,将解压之后的文件夹下的bin、include、lib文件拷贝到cuda安装目录对应的文件下。
检查是否安装成功:
WIN+R输入cmd,cd到安装目录的..\extras\demo_suite文件夹下:
分别执行bandwidthTest.exe和deviceQuery.exe,如果均出现PASS,则说明安装成功。
**第四步、pytorch安装**
这里强烈推荐用离线下载的方式安装,具体操作如下:
1.进入[pytorch官方网站](https://pytorch.org/),查找执行命令
上面没有CUDA9.0版本,因此进入[previous-versions网站](https://pytorch.org/get-started/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官网](https://www.jetbrains.com/pycharm/download/)下载本机版本,有个人版和社区版,个人版是免费的。
PyCharm安装完成之后需要解释器,创建一个新的项目,输入项目工作区、项目名,仍需注意的是解释器的选择。
选择Anaconda中python.exe所在路径即可。
最后,创建一个.py,运行一下,没有报错证明工程创建正确。
当然也可以搜一个需要使用GPU的代码,运行看是否运行成功。
到此,环境全部搭建结束了!!!