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的代码,运行看是否运行成功。
到此,环境全部搭建结束了!!!

https://editor.csdn.net/md/?articleId=107362672

posted @ 2020-07-15 21:02  cwhac  阅读(483)  评论(0编辑  收藏  举报