win+RTX30系显卡安装pytorch

在anaconda环境中安装

更换清华源：

​conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
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/conda-forge/

安装指令（可用）

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1

不需要另外安装cuda和cudnn了，anaconda已经帮我们装好了。

查看python cuda cudnn的版本：

import torch
print(torch.__version__)

import torch
print(torch.version.cuda)

import torch
print(torch.backends.cudnn.version())