我尝试了十几次来安装tensorflow-GPU,但都失败了,彻底的我遵循了成功安装者的指示。 我刚刚注意到我有2个GPU指示灯,这意味着我有2个GUP集?一台英特尔和一台Nvidia。
答案 0 :(得分:0)
使用以下步骤验证tensorflow-GPU的正确安装
您可以检查tensorflow是否使用GPU
import tensorflow as tf
print(tf.test.is_gpu_available())
要查明使用哪个设备,可以启用日志设备放置 像这样:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
您可以尝试将计算分配给gpu,看看是否有错误
import tensorflow as tf
with tf.device('/gpu:0'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
with tf.Session() as sess:
print (sess.run(c))
使用以下命令检查GPU使用情况
nvidia-smi