使用带有移动GPU(笔记本电脑)的Keras(Tensorflow后端)

时间:2017-09-26 00:52:33

标签: python tensorflow keras tensorflow-gpu

如果我只使用Tensorflow代码,则GPU使用率超过80%且温度上升非常多。但如果我使用Kers,使用率会下降到15%。此外,使用Keras无法达到GPU的最大时钟。

我尝试了980m,1070(笔记本电脑)和960m,但获得了相同的结果。 ANN和CNN都有相同的结果。

为什么在移动(笔记本电脑)GPU上使用Kers时,我的GPU利用率如此之低?在笔记本电脑上使用Keras时,我该怎么做才能确保获得最佳性能?

1 个答案:

答案 0 :(得分:0)

也许升级到Tensorflow 1.3。在TF 1.3中已经包含了Keras,并且不需要另外安装keras。

为了使用TF中包含的keras版本,请执行以下操作: 例如使用

from tensorflow.contrib.keras.python.keras.models import Model

from tensorflow.contrib.keras.python.keras.layers import Input, Conv2D, Dropout

而不是

from keras.models import Model
from keras.layers import Input, Conv2D, Dropout

也许这解决了问题!