1、使用nvidia-smi pmon 查看linux系统的gpu情况,如下:

显然是2张显卡,如何让它们都工作呢
2、keras提供了keras.utils import multi_gpu_model使用多个显卡的功能:
在原来的model基础上使用multi_gpu_model函数指定一下gpu个数即可:
model = multi_gpu_model(model, 2)
完整列子如下(如粗黑色字体描述):
root@deeplearning:/opt/soft/keras/examples# cat mnist_mlp_multi_gpu.py
最终运行的结果如下: