文章目录
1. 先安装驱动&CUDA
Ubuntu/Linux 安装CUDA、检测(PyTorch准备)
Ubuntu/Linux 安装GPU 驱动&检测(PyTorch准备)
2. 背景cuDNN v11.8
链接: 背景
2.1. 注册帐号
链接: 官网
2.2. 检查CUDA版本
nvcc -V
3. 下载安装(推荐)
sudo apt update && sudo apt upgrade -y
下载cuDNN压缩文件
CUDNN_TAR_FILE="cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz"
sudo wget https://developer.download.nvidia.com/compute/redist/cudnn/v8.7.0/local_installers/11.8/cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz
sudo tar -xvf ${CUDNN_TAR_FILE}
移动文件夹更改$PATH$
sudo mv cudnn-linux-x86_64-8.7.0