tensorflow查看深度学习项目是在CPU上 还是GPU上

import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = '99'

print(device_lib.list_local_devices())

 

©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页