在 TensorFlow 论坛上发现类似问题HERE 没有解决方案。
我在同一系统(本地 Windows 11)上运行 PyTorch 2.0.0 和 CUDA 11.8,没有任何问题,但在 WSL2 Ubuntu 中遇到 TensorFlow 2.12 未检测到 GPU 的问题。
系统详情:
已完成以下工作:
我要运行的代码:
import tensorflow as tf
print('TensorFlow Version : ',tf.__version__)
print('GPU Name : ',tf.config.list_physical_devices('GPU'))
Everything之后,系统因为某些原因无法获取cuDNN。 (环境名称不同,因为我卸载了 WSL 并再次从头开始尝试,仍然遇到同样的问题。)
如何解决这个问题并在 GPU 上运行 TensorFlow 2.12?
经过大量的试验和错误,这行得通!
'''
pip uninstall tensorflow
'''
'''
cd $CONDA_PREFIX/etc/conda/activate.d
nano env_vars.sh
'''
'''
CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
'''
'''
pip install --upgrade pip
pip install tensorflow==2.12.*
'''
第 9 步:重启系统
第 10 步:使用以下命令验证
'''
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
''' 您应该有一个带有以下行的输出
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]