1、版本选择
1）选择PyTorch版本：2.0.1
2）对应的CUDA版本：11.7或者11.8，选择较新的11.8
3）确定显卡是否支持，使用命令nvidia-smi查看显卡驱动版本及支持的CUDA，本人的为12
4）cuDNN版本
选择和CUDA版本对应的最新版本v8.9.1
2、下载
2.1 下载CUDA
各个版本CUDA下载
https://developer.nvidia.com/cuda-toolkit-archive
本人选择11.8
选择对应的系统版本
在终端中使用命令下载</