这几天给自己的笔记本重装pytorch，因为以前安装tensorflow的时候配置了CUDA和CUDNN，但是选pytorch版本的时候忘了自己安装的CUDA版本了（毕竟pytorch版本要求和CUDA的版本相一致才行），以此记录一下：

1. win+R，输入cmd打开命令行
2. 输入指令：nvidia-smi
   之后显示的表格里包含自己的CUDA版本。
   
   
   

另附pytorch官网针对不同版本的CUDA进行的版本组合推荐：

Pytorch最新版本的选择组合

Pytorch历史版本的选择组合

自己因为是CUDA11.1版本，而且比较习惯安装历史版本，自己最终选择的pytorch1.10版本，组合如下：

# CUDA 11.1
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html