測試程序：

import tensorflow as tfwith tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)) as sess:a = tf.constant(1)b = tf.constant(3)c = a + bprint('結果是：%d\n 值為：%d' % (sess.run(c), sess.run(c)))

輸出顯卡型號、內存等：

?

tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False))

?

allow_soft_placement=True 表示使用不能使用顯卡時使用cpu

log_device_placement=False 不打印日志，不然會刷屏
?

總結

以上是生活随笔為你收集整理的gpu版tensorflow测试的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。