Keras強制使用CPU

最近使用GPU來跑Keras模型速度很快，可是預測的時候加載的很是慢，估計是使用GPU的問題html GPU作並行運算效果優於CPU，可是在預測的時候不須要並行運算，若是再使用GPU的話致使加載時間太長post 所以訓練模型使用GPU，預測使用CPUspa   在導入Keras以前加入如下兩行代碼：code import os os.environ["CUDA_VISIBLE_DEVICES"]

>>阅读原文<<