简而来讲,我目前电脑中有三个可用的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

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐