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

Question

在Ubuntu 18.04上使用Tensorflow 2.0。跑步时

page.evaluate((selector) => {
  const element = document.querySelector(selector);
  element.click();
}, selector)

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

Answer 1

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

或

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

您也可以使用tf.config.experimental.list_physical_devices(device_type=None)。如果检测到可用的GPU，则说True。