tensorflow获取当前可用的GPU信息

简而来讲，我目前电脑中有三个可用的GPU，将返回['/ gpu：0'，'/ gpu：1'，'/ gpu：2']，可以使用以下代码检查所有设备列表：

from tensorflow.python.client import device_lib

def get_available_gpus():
    local_device_protos = device_lib.list_local_devices()
    return [x.name for x in local_device_protos if x.device_type == 'GPU']
print(get_available_gpus())

扫码关注

实用AI客栈

获取最新AI资讯与实战案例

小编微信号 : langu86