win10中anaconda创建虚拟环境配置pytorch1.8.0 gpu版 cunda9.0 cudnn7.3 并配置到pycharm上

关于anaconda、cuda和cudnn的安装教程可以看我的这篇博客，安装ptorch和tensorflow的前期过程是一样的，这里只详细讲解pytorch的安装过程。

---

安装pytorch

1、打开anaconda prompt，创建新的虚拟环境，命令如下：

conda create -n pytorch-gpu-1.8.0 python=3.6

2、切换到新的虚拟环境下，并查看list：

acativate pytorch-gpu-1.8.0
conda list

3、下载torch

这里使用的方法是将torch下载到本地，然后进行安装，避免版本不一致等问题，

下载地址

注意：

cu开头的才是版本，

torch后面的1.8.0是torch的版本号，

cp后面的3.6表示python的版本

4、安装torch

在anaconda prompt中，在新创建的虚拟环境下（我这里是pytorch-gpu-1.8.0环境下），通过cd命令将目录切换到torch的下载目录下，

然后输入安装命令：

pip install torch-1.8.0-cp36-cp36m-win_amd64.whl(这里是你下载下来的文件名)

回车等待安装完成即可。

5、测试安装成功

在anaconda prompt中输入python进入python环境，依次输入：

import torch
torch.__version__  #1.8.0
torch.cuda.is_available()  #True

如果返回结果如注释所示，那就说明pytorch（gpu）安装成功，否在失败。

配置到pycharm上

打开pycharm编辑器，在File下选择Settings，点击最右边的设置，选择Add，然后选择现有环境，将目录定位到新创建的anaconda虚拟环境下的python.exe，一路点击确定退出，即可完成配置。