TensorFlow查看GPU信息

查看是否有GPUpython import tensorflow as tf gpu_device_name = tf.test.gpu_device_name() print(gpu_device_name) 輸出web /device:GPU:0 GPU是否可用svg # 返回True或者False tf.test.is_gpu_available() from tensorflow.pyt
相關文章
相關標籤/搜索