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

【问题标题】：How to use specific GPU device for learning in keras R?如何在 keras R 中使用特定的 GPU 设备进行学习？
【发布时间】：2018-11-29 06:31:56
【问题描述】：

我是大斌。

我想在 keras R 而不是 python 中使用特定的 GPU 设备进行学习。因为当我尝试同时学习两个程序时，一个程序运行良好，而另一个程序不运行。我认为似乎另一个程序正在等待学习结束。所以，我有两个 GPU 1080 ti，我想为每个程序使用特定的设备。但是，multi_gpu_model 函数在使用两个以上设备的时候，对吧？

如果我可以使用 multi_gpu_model 功能使用一台设备进行学习。我如何知道参数“gpu=”的设备名称？

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

【问题讨论】：

标签： r keras

【解决方案1】：

默认情况下，gpu 名称为“/gpu:0”、“/gpu:1”...“/gpu:(n-1)”，如果您有 n 个 gpu。

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

【讨论】：

只使用一台设备可以使用 multi_gup_model 功能吗？我只想在每个程序中使用一台设备。
multi_gpu_model 让您选择多个 gpus 而不是单个，在脚本开头写入 os.environ["CUDA_VISIBLE_DEVICES"]="1" ，该脚本将只能查看和使用 / gpu:1 ，您可以对其他脚本执行相同的操作并更改数字，以便它们使用不同的 gpus。