打印出可被TensorFlow(TF)使用的设备
- 在ipython(python)中执行命令
from tensorflow.python.client import device_lib as _device_lib
devices = [x.name for x in local_device_protos if x.device_type == 'GPU']
for d in devices:
print(d)
- 如果不需要特别指定GPU,从打印出全部可用设备
from tensorflow.python.client import device_lib as _device_lib
devices = [x.name for x in local_device_protos]
for d in devices:
print(d)