我正在AWS的深度学习AMI（ubuntu 12）中使用keras训练NN。
如何确保Jupyter Notebook中的脚本正在使用GPU？
答案 0 :(得分：1)
在tensorflow中运行会话之前，您可以添加以下代码行进行检查：
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
有关在tensorflow中使用GPU的更多详细信息，您可以在tensorflow官方网站上查看：https://www.tensorflow.org/guide/using_gpu