Tensorflow中查看gpu是否可用

使用tf.test.is_gpu_available()函數可直接返回

import tensorflow as tf tf.test.is_gpu_available()

?

Tensorflow測試程序

# Python import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello)) 創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的Tensorflow中查看gpu是否可用的全部內容，希望文章能夠幫你解決所遇到的問題。

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