如何在WSL2上安装适配TensorFlow 2的CUDA 11.2与cuDNN 8.1？

问题说明 #

此前已在WSL2的Ubuntu 20.04.5 LTS上成功安装CUDA 11.8，但TensorFlow 2仅支持特定版本的CUDA与cuDNN组合，因此需要安装CUDA 11.2和cuDNN 8.1来适配开发需求。

设备配置 #

• 操作系统：Windows 10 22H2（支持WSL2）
• 处理器：AMD Ryzen 7 5800H
• 内存：32GB
• 显卡：NVIDIA GeForce RTX 3070 Laptop GPU
• WSL2发行版：Ubuntu 20.04.5 LTS

安装步骤 #

1. 卸载现有CUDA版本（可选但推荐） #

如果之前安装过其他CUDA版本，先彻底卸载：

sudo apt-get purge cuda*
sudo apt-get autoremove
sudo rm -rf /usr/local/cuda*

2. 安装CUDA 11.2 #

执行以下命令完成CUDA 11.2的安装：

# 下载并配置优先级文件
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600

# 下载CUDA 11.2本地安装包
wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-wsl-ubuntu-11-2-local_11.2.0-1_amd64.deb

# 安装并配置密钥
sudo dpkg -i cuda-repo-wsl-ubuntu-11-2-local_11.2.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-11-2-local/cuda-*-keyring.gpg /usr/share/keyrings/

# 更新源并安装CUDA 11.2
sudo apt-get update
sudo apt-get -y install cuda-11-2

3. 安装cuDNN 8.1 #

执行以下命令完成cuDNN 8.1的安装：

# 下载cuDNN 8.1.0适配CUDA 11.2的压缩包
wget https://developer.download.nvidia.com/compute/machine-learning/cudnn/secure/8.1.0.77/11.2_20210127/cudnn-11.2-linux-x64-v8.1.0.77.tgz

# 解压并复制文件到CUDA目录
tar -xzvf cudnn-11.2-linux-x64-v8.1.0.77.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.2/lib64
sudo chmod a+r /usr/local/cuda-11.2/include/cudnn*.h /usr/local/cuda-11.2/lib64/libcudnn*

4. 配置环境变量 #

编辑个人shell配置文件（以.bashrc为例）：

nano ~/.bashrc

在文件末尾添加以下内容：

export PATH=/usr/local/cuda-11.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64:$LD_LIBRARY_PATH

保存并退出后，执行命令使配置生效：

source ~/.bashrc

验证安装 #

执行以下命令确认安装成功：

# 查看CUDA版本
nvcc --version

# 查看cuDNN版本
cat /usr/local/cuda-11.2/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

内容的提问来源于stack exchange，提问作者aladin