查看tensorflow是否支持GPU，以及測試程序

• 2019 年 10 月 28 日
• 筆記

版權聲明：本文為博主原創文章，遵循 CC 4.0 BY-SA 版權協議，轉載請附上原文出處鏈接和本聲明。

本文鏈接：https://blog.csdn.net/weixin_36670529/article/details/102759936

測試程序

# Python  import tensorflow as tf  hello = tf.constant('Hello, TensorFlow!')  sess = tf.Session()  print(sess.run(hello))

是否支持GPU

import tensorflow as tf  sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))  >>>  Device mapping:  /job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce GTX 960M, pci bus id: 0000:02:00.0, compute capability: 5.0