keras gpu的問題

官方這樣說道 如果是tensorflow、cntk作爲後端，那麼會自動檢測 theano可以手動設置 但是我這的速度，讓人懷疑並沒有運行在gpu上（一張圖，運行時間近10秒） 嘗試運行時指定 CUDA_VISIBLE_DEVICES=0 python demo.py 運行時間無變化 顯式設置backend import keras.backend.tensorflow_backend as KTF

>>阅读原文<<