windows + anaconda + pycharm + nvidia + cuda + cudnn + pytorch 安装教程

要学pytorch了耶，这不得安装一把子环境？

查看显卡信息

操作：开始 —— 右键 —— 设备管理器 —— 显示适配器。

在这里插入图片描述

查看cuda版本

操作：右下角NVIDIA —— 右键 —— NVIDIA控制面板 —— 系统信息 —— 组件。

在这里插入图片描述

安装cuda

去CUDA官网下载cuda10.2版本。

在这里插入图片描述
下载后双击exe文件，更改目录，解压。

在这里插入图片描述
解压完成后会自动进入安装流程。

在这里插入图片描述
按照安装提示默认安装即可。

在这里插入图片描述

安装中。
在这里插入图片描述

检查是否安装成功。

nvcc -V

在这里插入图片描述
成功嘞！

安装cudnn

去CUDNN官网下载cuda10.2版本对应的cudnn。

在这里插入图片描述

下载之后，解压缩，将CUDNN压缩包里面的bin、include、lib文件直接复制到CUDA的安装目录下（C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2）。

这里会遇到管理员权限问题。

在这里插入图片描述
在”为所有当前项目执行此操作“前点击对号再点击继续。

在这里插入图片描述

配置cuda环境变量

已经有两个了。
在这里插入图片描述
还要添加三个。

一开始找不到CUDA Samples，但是开始菜单可以看到。

在这里插入图片描述
打开即可。

后来发现，系统变量的PATH里有bin和libnvvp就可以了。

在这里插入图片描述

配置cudnn环境变量

在这里插入图片描述

安装pytorch

在conda里面查看python版本。
在这里插入图片描述
在conda里面创建虚拟环境。

wxm_pytorch是虚拟环境名字。

conda create -n wxm_pytorch python=3.7

在这里插入图片描述
点击y。

在这里插入图片描述

激活虚拟环境。

conda activate wxm_pytorch

在这里插入图片描述

进入pytorch官网选择配置项。

conda安装

在这里插入图片描述

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

在这里插入图片描述
输入y，等待安装。

在这里插入图片描述
下好啦。

在这里插入图片描述
下错了。

先删除这个失败的虚拟环境。

在这里插入图片描述
重来。

pip安装

在这里插入图片描述

pip3 install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

下好了。

在这里插入图片描述

测试一下：

在这里插入图片描述
开心心。

在这里插入图片描述
看到torch了。

第一个DEMO

在这里插入图片描述
终于！！！