您好,登錄后才能下訂單哦!
這篇文章主要介紹了已安裝tensorflow-gpu但keras無法使用GPU加速怎么辦,具有一定借鑒價值,感興趣的朋友可以參考下,希望大家閱讀完這篇文章之后大有收獲,下面讓小編帶著大家一起了解一下。
問題
我們使用anoconda創建envs環境下的Tensorflow-gpu版的,但是當我們在Pycharm設置里的工程中安裝Keras后,發現調用keras無法使用gpu進行加速,且使用的是cpu在運算,這就違背了我們安裝Tensorflow-gpu版初衷了。
原因
因為我們同時安裝了tensorflow和tensorflow-gpu(在…Anaconda3\envs\fyy_tf\Lib\site-packages中可以找到他們的文件夾),使用keras時會默認調用tensorflow,從而無法使用GPU進行訓練。
解決方法
同時卸載tensorflow、tensorflow-gpu 和 keras(可以直接去…Anaconda3\envs\fyy_tf\Lib\site-packages下刪除他們的文件夾);
然后,再在Pycharm設置中使用小加號安裝tensorflow-gpu 和 keras。
最后就可以使用keras進行gpu加速。
感謝你能夠認真閱讀完這篇文章,希望小編分享的“已安裝tensorflow-gpu但keras無法使用GPU加速怎么辦”這篇文章對大家有幫助,同時也希望大家多多支持億速云,關注億速云行業資訊頻道,更多相關知識等著你來學習!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。