判断TensorFlow安装的是cpu,还是gpu版本，以及TensorFlow版本号

import tensorflow as tf
import os
from tensorflow.python.client import device_lib

if __name__=="__main__":
    print(tf.__version__)
    print(device_lib.list_local_devices())
版权声明：本文为cute_leaf原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。