問(wèn)題
我們使用anoconda創(chuàng)建envs環(huán)境下的Tensorflow-gpu版的,但是當(dāng)我們?cè)赑ycharm設(shè)置里的工程中安裝Keras后,發(fā)現(xiàn)調(diào)用keras無(wú)法使用gpu進(jìn)行加速,且使用的是cpu在運(yùn)算,這就違背了我們安裝Tensorflow-gpu版初衷了。
原因
因?yàn)槲覀兺瑫r(shí)安裝了tensorflow和tensorflow-gpu(在…Anaconda3/envs/fyy_tf/Lib/site-packages中可以找到他們的文件夾),使用keras時(shí)會(huì)默認(rèn)調(diào)用tensorflow,從而無(wú)法使用GPU進(jìn)行訓(xùn)練。
解決方法
同時(shí)卸載tensorflow、tensorflow-gpu 和 keras(可以直接去…Anaconda3/envs/fyy_tf/Lib/site-packages下刪除他們的文件夾);
然后,再在Pycharm設(shè)置中使用小加號(hào)安裝tensorflow-gpu 和 keras。
最后就可以使用keras進(jìn)行g(shù)pu加速。
以上這篇已安裝tensorflow-gpu,但keras無(wú)法使用GPU加速的解決就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持武林站長(zhǎng)站。
新聞熱點(diǎn)
疑難解答
圖片精選