日々精進

新しく学んだことを書き留めていきます

2016-11-03から1日間の記事一覧

KerasがGPUを使ってくれなくて困った問題

AI

原因は以下。 ・Kerasで複数GPU対応をした。そのとき、with tf.device("/cpu:0"): を追加してCPUでモデルを構築するようにした。 ・その時はmulti_gpu_modelを使って複数GPUで動いた気がしたが、その後1GPUマシンで動かしたところGPUを使わなくなっていた。…