如何在Keras的fit_generator（）中训练时使用多个GPU？

Question

我当前正在使用keras's model.fit_generator()函数，并且我有两个GPU。但是，当我运行代码时，仅使用了1个GPU。我想知道如何使用另一个GPU。

我已经阅读了keras文档，并猜测设置use_multiprocessing=True应该可以解决问题，但同样，只有一个GPU正在运行。

我也尝试过os.environ["CUDA_VISIBLE_DEVICES"] = '0, 1'，但这也不起作用

有人可以帮我吗？