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

版权声明：本文为博主原创文章，转载请标明出处。 https://blog.csdn.net/chuan403082010/article/details/89070421

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

print(device_lib.list_local_devices())