Question

我刚刚启动了一个试图训练模型的AWS P2实例。然而，似乎使用CPU来训练不是GPU。如何强制它使用GPU而不是CPU进行训练？

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

我收到消息＆＃34;无法加载本机TensorFlow运行时。＆＃34;

然后我改变了

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

它的训练非常缓慢，似乎正在使用cpu。

Answer 1

看起来答案是添加一个gpus标志！

python cnn_homework_solution.py --gpus 0,1

使用GPU而不是CPU在AWS上训练

1 个答案: