2017-10-01 22 views
0

モデルをトレーニングしようとしているAWS P2インスタンスを起動しました。しかし、それはGPUを訓練するためにCPUを使用しているようです。 CPUではなくGPUを使って訓練をするにはどうすればいいですか?GPUを使用しないAWSのトレーニングCPU

$ nano ~/.keras/keras.json says this 
{
 "image_dim_ordering": "th",
  
"epsilon": 1e-07, 

 "floatx": "float32",
  
"backend": "tensorflow"
 } 

「TensorFlowネイティブの読み込みに失敗しました」というメッセージが表示されます。

私はそれはしかし、非常にゆっくりと、訓練だとCPUを使っているように見える

$ nano ~/.keras/keras.json says this 
{
 "image_dim_ordering": "th",
  
"epsilon": 1e-07, 

 "floatx": "float32",
  
"backend": "theano"
 } 

を変更しました。

答えて

0

答えがgpusフラグを追加しているようです。

python cnn_homework_solution.py --gpus 0,1