Tensorflow V1未检测到GPU

pvabu6sv 于 2022-12-13 发布在其他

关注(0)|答案(1)|浏览(208)

我目前使用的是Tf 1.1.0。

print(device_lib.list_local_devices())

它只显示CPU而不显示GPU。当我用tf 2.x.x运行相同的命令时，CPU和GPU都显示了。有没有办法让tf v1检测我的GPU，因为我不愿意切换到tf v2
我的模型正在CPU上训练，因为tf未检测到GPU

1条答案

如果您不想切换到TensorFlow 2.x，可以尝试更新GPU驱动程序，并指定运行TensorFlow时要使用的GPU。您可以使用CUDA_VISIBLE_DEVICES环境变量来执行此操作。例如：

CUDA_VISIBLE_DEVICES=0 python your_script.py

这将指示TensorFlow在运行脚本时使用第一个GPU（索引为0）。您可以更改索引以指定不同的GPU。