TensorFlow 2.16.1无法识别GPU的问题求助

针对Ubuntu 23.10下TensorFlow 2.16.1无法检测GPU的已知bug，可尝试以下解决方案：

• 降级TensorFlow版本
  直接降级到2.16.0或2.15.x版本，该版本修复了GPU检测问题：

  # 降级到2.16.0
  pip install tensorflow[and-cuda]==2.16.0 --force-reinstall
  # 或降级到2.15.0
  pip install tensorflow[and-cuda]==2.15.0 --force-reinstall
  

• 手动指定CUDA库路径（临时修复）
  运行Python前设置环境变量，让系统找到TensorFlow自带的CUDA依赖库：

  export LD_LIBRARY_PATH=$CONDA_PREFIX/lib/python3.11/site-packages/nvidia/cuda_nvrtc/lib:$CONDA_PREFIX/lib/python3.11/site-packages/nvidia/cuda_runtime/lib:$CONDA_PREFIX/lib/python3.11/site-packages/nvidia/cudnn/lib:$LD_LIBRARY_PATH
  

  也可以在Python代码开头添加路径配置：

  import os
  lib_path = os.environ['CONDA_PREFIX']
  os.environ['LD_LIBRARY_PATH'] = f"{lib_path}/lib/python3.11/site-packages/nvidia/cuda_nvrtc/lib:{lib_path}/lib/python3.11/site-packages/nvidia/cuda_runtime/lib:{lib_path}/lib/python3.11/site-packages/nvidia/cudnn/lib:{os.environ.get('LD_LIBRARY_PATH', '')}"
  import tensorflow as tf
  

• 切换到conda官方TensorFlow包
  卸载pip安装的版本后，改用conda安装适配性更好的GPU版本：

  pip uninstall tensorflow -y
  conda install tensorflow-gpu==2.16.1
  

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