如何在tensorflow中获取当前可用的GPU？

fantomas

简而言之，我想要机器中有两个可用的GPU，它将返回['/ gpu：0'，'/ gpu：1']。


可以使用以下代码检查所有设备列表：

1. <blockquote>from tensorflow.python.client import device_lib

复制代码

1. from tensorflow.python.client import device_lib
   
2. 
   
3. def get_available_gpus():
   
4.     local_device_protos = device_lib.list_local_devices()
   
5.     return [x.name for x in local_device_protos if x.device_type == 'GPU']

复制代码

lsh手机认证

这是做GPU并发
的必备功能

不错。
TF是完备的庞然大物