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

Question

我有两个 GPU 设备 GTX 1080ti。 我在 Rstudio 中使用 keras 进行工作。 所以，我想为每个脚本使用特定的 GPU 设备。 我听说有人建议使用

with(tf$device("/:gpu1")

。 但是，我使用的是 R 而不是 python。 我找不到 tf 函数和 device 函数。 另一个建议是使用

os.environ["CUDA_VISIBLE_DEVICES"]="1"

。 但是，我找不到 os.environ 函数。

请帮助我任何可以帮助的人。 因此，请告诉我有关此问题的代码的更多详细信息。 谢谢。

Answer 1

Sys.setenv(CUDA_VISIBLE_DEVICES="1")