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

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