tensorflow获取当前可用的GPU信息

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())