【AI】[keras]GPUではなくCPUを使用するコード | 夕湖津のブログ

夕湖津のブログ

問題解決に役立つ情報の提供を目指します

■結論

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "1"  #これはCPUパフォーマンス最適化用なので任意

 

■経緯

kerasで簡単な学習をさせる時に、カーネルがすぐに逝くので、

バッチサイズを減らしてみましたがだめ。

GPUを大量使用していることが分かったので、

CPUのみ使用する方法でうまくいきました。

 

※kerasのバージョンは、2.10.0