WSL2 Ubuntu中TensorFlow无法识别GPU及缺失库问题求助

问题根源 #

1. TensorFlow版本与当前CUDA版本不匹配：你Windows端nvidia-smi显示的CUDA 12.0需要对应TensorFlow 2.13及以上版本，旧版TF依赖更低版本的CUDA库（比如11.0），导致加载失败。
2. WSL2环境未正确安装适配的CUDA Toolkit：nvidia-smi显示的是Windows端的CUDA驱动版本，WSL2需要单独安装适配Linux的CUDA Toolkit才能让TensorFlow调用GPU。

修复步骤 #

1. 确认并调整TensorFlow版本

首先检查当前TensorFlow版本：

python -c "import tensorflow as tf; print(tf.__version__)"

• 如果版本低于2.13，升级到兼容CUDA 12.0的版本：

  pip install --upgrade tensorflow>=2.13
  

• 若需使用旧版TensorFlow，则对应安装兼容的CUDA版本（比如TF2.12对应CUDA11.8）。

2. 在WSL2 Ubuntu中安装适配的CUDA Toolkit

卸载可能存在的旧CUDA组件：

sudo apt-get purge nvidia* cuda*
sudo apt-get autoremove && sudo apt-get clean

添加NVIDIA WSL专属软件源：

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
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/ /"

安装CUDA 12.0（对应TF2.13+）：

sudo apt-get update && sudo apt-get install cuda-12-0

配置环境变量，编辑~/.bashrc（或~/.zshrc，根据你的shell）：

echo 'export PATH=/usr/local/cuda-12.0/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc

生效环境变量：

source ~/.bashrc

3. 安装匹配版本的cuDNN

cuDNN是TensorFlow GPU加速必需的库，安装与CUDA12.0兼容的版本：

sudo apt-get install libcudnn8=8.9.4.25-1+cuda12.0
sudo apt-get install libcudnn8-dev=8.9.4.25-1+cuda12.0

4. 验证修复

• 检查CUDA Toolkit版本：

  nvcc --version
  

  应显示release 12.0相关内容。
• 重新运行你的测试脚本：

  import tensorflow as tf
  import os
  os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
  
  print(tf.reduce_sum(tf.random.normal([1000, 1000])))
  print(tf.config.list_physical_devices('GPU'))
  

  此时应能输出GPU设备列表，且无CUDA库加载警告。

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