2
我已经编写了一些基本的Deep Learning
代码,其中有2个LSTM
层。我使用Keras
和Theano
作为我的后端。我的机器上的这段代码在AWS
上比在AWS
上的另一台机器上花费的时间太长。在运行速度更快的机器上,每个纪元需要640秒,而在运行速度较慢的机器上,每个纪元需要超过10,000秒。确保Python代码是否在GPU或CPU上运行
我开始认为较慢机器上的代码没有在GPU上运行。两台机器上运行的代码完全相同。机器配置也是一样的。
看起来像是Theano
安装在较慢的机器上。我跑了下面的代码,并得到了结果:
有没有一种方法来检查,如果我的代码是在GPU或CPU上运行?
在这方面的任何帮助将不胜感激。
TIA。
编辑
按照从@Marcin提醒,我添加以下代码:
但是当我运行下面的代码,我仍然得到Used the cpu
结果:
非常感谢回复。我运行了你的一段代码,并且确实得到了'使用cpu'。你会碰巧知道如何改变它以使用'GPU'? – Patthebug
你能打印出'theano'标志的结果吗? –
哦,上帝 - 看来你还没有安装'Theano'。这也可以解释巨大的训练时间。 –