代码之家  ›  专栏  ›  技术社区  ›  Mauro Gentile

如何检查我是否在AWS上使用GPU?

  •  0
  • Mauro Gentile  · 技术社区  · 6 年前

    我正在AWS的一个深度学习am I(ubuntu 12)中使用keras训练NN。

    如何确保Jupyter笔记本中的脚本使用GPU?

    1 回复  |  直到 6 年前
        1
  •  1
  •   Richard Wan    6 年前

    在tensorflow中运行会话之前,可以添加这行代码来检查:

    sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
    

    https://www.tensorflow.org/guide/using_gpu