如何在keras R中使用特定的GPU设备进行学习？

Question

我是达彬。

我想使用特定的GPU设备在keras R中而不是python中进行学习。 因为当我尝试同时学习两个程序时，一个程序可以很好地工作，而另一个程序却不能工作。 我认为似乎另一个程序正在等到学习结束。 因此，我有两个GPU 1080 ti，我想为每个程序使用特定的设备。 但是，multi_gpu_model函数使用当使用两个以上的设备时，对吗？

如果我可以使用multi_gpu_model函数来使用一台设备进行学习。 我怎么知道参数“ gpu =” name的设备名称？

而且，如何为每个程序使用特定的设备？

Answer 1

如果您有n个GPU，默认的GPU名称默认为“ / gpu：0”，“ / gpu：1” ...“ / gpu：（n-1）”。

对于multi_gpu_model函数的gpus参数，您可以传递gpu名称列表而不是整数。