def count_gpus():
from tensorflow.python.client import device_lib
count = 0
for device in device_lib.list_local_devices():
if device.device_type == "GPU":
count+=1
return count
微信
公众号
扫码关注公众号