1. 程式人生 > >Keras無法呼叫tensorflow-gpu的解決方案

Keras無法呼叫tensorflow-gpu的解決方案

今天用keras訓練時發現記憶體佔用率出奇的高,而且視訊記憶體佔用率出奇的低,原來keras沒有用gpu訓練。
通過

pip list

看到同時安裝了tensorflow和tensorflow-gpu,keras預設呼叫了tensorflow
解決辦法
同時解除安裝tensorflow,tensorflow-gpu,keras
再安裝tensorflow-gpu和keras就可以解決