tensorflow获取当前可用的GPU信息

tensorflow获取当前可用的GPU信息_第1张图片 GPU云服务器(支持Jupyter)

内置Jupyter、Pycharm的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())

 

你可能感兴趣的:(tensorflow)