使用 tensorflow 查询 GPU 设备列表

from tensorflow.python.client import device_lib
def get_all_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_all_gpus())

注意：docker 容器开启 privileged 参数后可以访问所有的 GPU 设备，不管是否传入 Devices 参数，建议多租户隔离的云服务都禁用 privileged 功能。