win10下CUDA11.1+CUDNN8.0+Pytorch1.8.0安装教程

win10下CUDA11.1+CUDNN8.0+Pytorch1.8.0安装教程

cuda和cudnn10.1-11.1链接：
链接：https://pan.baidu.com/s/1HWVxNYXg94-9FEPer6BdaQ?pwd=da0z
提取码：da0z

cuda11.1安装

1.执行cuda.exe

2.不要改路径，就默认
3.
4.不点VS那个选项，并且Driver components和Other components也不勾选。
5.安装默认的位置
6.完成

配置环境变量

电脑→属性→高级系统设置→环境变量
在系统变量里多了这两项

CUDNN安装

把压缩包解压，打开后会出现这样的目录。
之前全选将他们都复制到CUDA目录下，C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1，这是默认的路径。

CUDA和CUDNN安装完测试

在路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite中，用cmd运行bandwidthTest.exe和deviceQuery.exe。
如果得到两个PASS就证明成功了，如图。

Pytorch1.8.0安装

pytorch官网、pytorch旧版本

这里我们要安装的是pytorch1.8.0版本，使用这个指令

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

anaconda使用

anaconda换源

切换源来提高下载速度，命令如下：

conda config --remove-key channels
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple

这个指令直接在cmd上就可以运行，如果报错就去环境变量里仔细看一下，anaconda的环境有没有写入。

创建虚拟环境

首先创建Python3.x的虚拟环境(这里我用3.8的），命令如下：
这里mmaction2，是我虚拟环境的名字，你们可以替换成自己想要的名字。

conda create -n mmaction2 python==3.8.5
conda activate mmaction2

第一句是创建虚拟环境，第二句是激活虚拟环境。

pytorch安装——cmd命令安装

进入虚拟环境后，运行命令：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

稍等一会后，输入Y然后等候下载。

完成。

pytorch安装——离线安装

在cmd中进入虚拟环境（激活虚拟环境然后复制、粘贴）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

今天9.29号我用命令安装真的好慢好慢QAQ，强推离线安装

cmd命令安装pytorch

ps：在cmd中输入以下命令（最后三条代码，按实际需求选一个安装即可）

#创建虚拟环境（名字为open-mmlab Python的版本是3.8）名字和版本是根据自己需求改的 这里只是举例子
conda create -n open-mmlab python=3.8 -y
#激活虚拟环境
conda activate open-mmlab
#安装cuda为11.1版本所对应的pytorch(官网命令中-c pytorch / -c conda-forge要删除，否则从官网路径下载会很慢）
#这里cudatoolkit=xx.x 要注意，因为我的cuda是11.1 所以这边是11.1  这边的xx.x要根据你的cuda写
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1
#如果电脑没有显卡，跑cpu的话就不跑上面的代码 
#跑这下面所示的（这个命令是安装最新版本的pytorch-cpu版本）
conda install pytorch torchvision torchaudio cpuonly

离线安装pytorch（下载发现这个方法更快）

！！！如果用cmd命令安装pytorch太慢了，你可以使用将安装包下载到本地（这边如果你需要其他版本的可以在这个链接下面找）
北京外国语的源pytorch的网站，选择你pytorch所对应的版本（这边要根据你虚拟环境的python版本、电脑cuda的版本！！一定要一一对应）
如果你需要的和我一样 可以直接点击下载
pytorch-1.8.0-py3.8_cuda11.1_cudnn8_0的下载链接
如果pytorch这个安装不成功的，试试另外一个方法：下载pytorch.whl文件链接。他的安装方式和.tar.bz2一样都是pip install xxxxx.whl / xxx.tar.bz2
torchaudio-0.8.0-py38.tar.bz2的下载链接
torchvision-0.9.0-py38_cu111.tar.bz2的下载链接
下载好后，打开cmd，进入你的虚拟环境然后运行

#这里是你下载的包的路径，同理torchvision、torchaudio也是这样安装的
#这边应该是你的路径，在win上就 例如d:\users\zlx\downloads\pytorch-1.8.0-py3.8_cuda11.1_cudnn8_0.tar.bz2 
conda install xxxx.tra.bz2
conda install xxxx.whl

安装好了之后，来测试一下GPU是否安装成功，出现下图就代表成功了：

代码：

python
import torch
print(torch.__version__)
print(torch.cuda.is_available())