最近使用GPU來跑Keras模型速度很快，但是預測的時候加載的非常慢，估計是使用GPU的問題
GPU做並行運算效果優於CPU，但是在預測的時候不需要並行運算，如果再使用GPU的話導致加載時間太長
因此訓練模型使用GPU，預測使用CPU
在導入Keras之前加入以下兩行代碼：
import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
既可以使用CPU來預測模型
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。