检测tensorFlow的GPU版本安装成功方法

打开终端,切换到tensorflow环境下

import tensorflow as tf
print('GPU',tf.test.is_gpu_available())
a = tf.constant(2.)
b = tf.constant(4.)
print(a * b)

输出结果为:
检测tensorFlow的GPU版本安装成功方法_第1张图片
表示tensorflow的gpu版本安装成功。

你可能感兴趣的:(pytorch基础,tensorflow,深度学习,机器学习)