[英]Can Keras with Tensorflow backend be forced to use CPU or GPU at will?
[英]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.