問題
我們使用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加速的解決就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答