簡體   English   中英

將Keras(Tensorflow后端)與移動GPU(筆記本電腦)結合使用

[英]Use Keras (Tensorflow backend) with mobile GPU (laptop)

如果僅使用Tensorflow代碼,則GPU利用率超過80%,並且溫度會大大升高。 但是,如果我使用Kers,使用率會下降到15%。 另外,使用Keras不能達到GPU的最大時鍾。

我嘗試使用980m,1070(筆記本電腦)和960m,但獲得了相同的結果。 ANN和CNN的結果相同。

在移動(筆記本)GPU上使用Kers時,為什么我的GPU利用率如此之低? 在筆記本電腦上使用Keras時,我該怎么做才能確保獲得最佳性能?

也許升級到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

也許可以解決問題!

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM