可用以下代码查看可使用设备情况
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
`如果只有CPU,会出现类似以下图示:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices
可用以下代码查看可使用设备情况
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
`如果只有CPU,会出现类似以下图示:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices