Keras + Tensorflow：对多个GPU的预测

Keras + Tensorflow：对多个GPU的预测

我创建了一个简单的示例来展示如何在多个GPU上运行keras模型。基本上，将创建多个进程，并且每个进程都拥有一个GPU。要指定正在处理的GPU
ID，设置环境变量CUDA_VISIBLE_DEVICES是一种非常简单的方法（os.environ [“
CUDA_VISIBLE_DEVICES”]）。希望这个git repo可以帮助您。

https://github.com/yuanyuanli85/Keras-Multiple-Process-
Prediction