AttributeError：模块“tensorflow_core._api.v2.config”没有属性“list_physical_devices”

Question

我在 Ubuntu 18.04 上使用 Tensorflow 2.0。正在运行

tf.config.list_physical_devices('GPU')

我收到上述错误。解决方法是什么？

Answer 1

运行tf.config.experimental.list_physical_devices('GPU')以查看所有 GPU

或者

运行tf.config.experimental.list_physical_devices(device_type=None)以查看所有设备

您也可以使用tf.test.is_gpu_available(). 如果检测到可用的 gpu，它会显示 True。