我看到了2个GPU指示灯,这可能是我一直无法安装tensorflow-GPU的原因吗?

时间:2019-09-11 17:36:42

标签: tensorflow anaconda gpu

我尝试了十几次来安装tensorflow-GPU,但都失败了,彻底的我遵循了成功安装者的指示。 我刚刚注意到我有2个GPU指示灯,这意味着我有2个GUP集?一台英特尔和一台Nvidia。

可能是原因吗?如果是这样,我该如何踢出合适的人(英伟达)? 谢谢你的建议。 2

1 个答案:

答案 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