使用TensorFlow2训练聊天机器人时GPU无法调用问题求助

解决方法 #

1. 修正TensorFlow版本不一致问题 #

脚本显示当前环境实际运行的是TensorFlow 2.10.0，与你声明的2.12.0不符，先清理旧版本：

pip uninstall -y tensorflow tensorflow-gpu

2. 匹配TensorFlow与CUDA/cuDNN版本 #

TensorFlow对CUDA版本有严格兼容性要求：

• 若保留CUDA 10.1，只能安装TensorFlow 2.3.x~2.4.x版本，例如：pip install tensorflow==2.4.0
• 若要使用TensorFlow 2.12.0，必须升级CUDA到11.8，同时搭配cuDNN 8.6.0（适配CUDA 11.x系列）

3. 更新NVIDIA驱动 #

GTX 1050 TI支持CUDA，确保驱动是适配Windows 11的最新版本，可通过NVIDIA GeForce Experience或官网下载安装。

4. 配置系统环境变量 #

安装CUDA和cuDNN后，将以下路径添加到系统PATH环境变量：

• CUDA的bin目录：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v[你的版本号]\bin
• cuDNN解压后的bin目录（建议将cuDNN文件直接复制到CUDA对应目录，或单独添加其bin路径）

5. 重新验证配置 #

完成上述步骤后，再次运行检测脚本，若输出显示TensorFlow was built with CUDA support且Num GPUs Available: 1，说明配置成功。

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