GPU未被Tensorflow V1检测到

Question

我目前使用的是Tf 1.1.0。我尝试使用以下命令

列出可用的设备

print(device_lib.list_local_devices())

只显示CPU，不显示GPU。当我用tf 2.x运行相同的命令时。显示CPU和GPU。有没有办法让tf v1检测我的GPU，因为我不愿意切换到tf v2

我的模型正在CPU上进行训练，因为tf没有检测到GPU

Answer 1

如果你不愿意切换到TensorFlow 2。x，你可以尝试更新你的GPU驱动程序，并指定运行TensorFlow时使用哪个GPU。您可以使用CUDA_VISIBLE_DEVICES环境变量来做到这一点。例如:

CUDA_VISIBLE_DEVICES=0 python your_script.py

这会告诉TensorFlow在运行脚本时使用第一个GPU(索引为0)。你可以改变索引来指定不同的GPU。