带RTX3090的win10电脑安装pytorch1.8 GPU版

系统为win10，显卡为RTX3090
CUDA

1. 下载和安装

下载链接：https://developer.nvidia.com/cuda-downloads

依次选择：Windows - x86_64 - 10 - exe(local)，点击 Download。由于RTX3090为比较新的显卡的（目前RTX3090Ti是最新的），下载的时候直接下载最新的 cuda 11.7

避坑：复制下载链接后用迅雷下载，比直接下载快数倍。

下载后双击，一路下一步就好。

2. 测试

如果安装成功，在 cmd 里输入：

nvcc -V

如果显示如下信息，表明安装成功。

如果没有的话，你有可能需要调整路径。请网上搜索。

cuDNN

1. 下载

下载网址：https://developer.nvidia.com/rdp/cudnn-download

网上教程建议在的情况下操作，注册时的验证码是谷歌的。

其实我只是用QQ邮箱注册的，使用时也没用遇到任何问题。总之可以先试一试QQ和163这类邮箱，不行再考虑outlook邮箱等国际邮箱。

你需要注册账号后才能继续。
由于RTX3090为比较新的显卡的（目前RTX3090Ti是最新的），下载的时候直接下载了最新的 cuda 11.7，相应的cudnn也使用了最新版的cudnn

2. 安装

说是安装，其实是复制文件。

下载后是一个压缩文件夹，解压文件夹，将里面的 3 个文件夹：bin，include，lib 里面的内容分别放入 CUDA 安装位置的对应文件夹。

CUDA 安装的位置在：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

不会出现覆盖的警告，因为是新增的文件。

pytorch

1. 下载与安装

避坑：务必在完成以上步骤后才进行。

网址：https://pytorch.org/get-started/locally/，根据实际情况选择。虽然我们安装的是 CUDA 11.7，这里只有 11.1，但是没有关系，是可以兼容的。

参考官网（https://pytorch.org/get-started/locally/）和安装的CUDA版本，在开始菜单找到Anaconda Prompt并右键选择管理员运行，拷贝官网的pip安装命令并修改CUDA版本（CUDA11.1版本为例），在打开的命令行中输入如下命令。

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html 

等待安装完成，输入ipython后，依次输入

import torch
import torchvision
print(torch.cuda.is_available())

等待屏幕返回True，说明GPU版本的PyTorch安装完成。