tensorflow查看是否有GPU可用

查看是否有GPU可用

from tensorflow.python.client import device_lib

# 列出全部的本地机器设备
local_device_protos = device_lib.list_local_devices()
# 打印
print(local_device_protos)

# 只打印GPU设备
[print(x) for x in local_device_protos if x.device_type == 'GPU']

tensorflow使用GPU

import os 
os.environ["CUDA_VISIBLE_DEVICES"] = "1"   # 数字为gpu编号