- 创建虚拟环境
- 在虚拟环境中下载对应版本torch、torchvision、torchaudio https://download.pytorch.org/whl/torch_stable.html
- CUDA官网查找对应版本cuda、cudnn(cuDNN Archive | NVIDIA Developer)教程
- Driver不选择;
- 设置 CUDA 11.1 的环境变量,改变系统当前所用cuda
export CUDA_HOME=/usr/local/cuda-11.1
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH