如何从系统RAM扩展tensorflow的GPU内存

Question

我想使用mscoco数据集code使用faster_rcnn_with resnet101训练谷歌对象检测。我只使用了10,000张图像用于训练目的。我使用了图形：GeForce 930M / PCIe / SSE2。 NVIDIA驱动程序版本：384.90。这是我的GeForce的图片。

我有8Gb RAM，但在tensorflow gpu中显示1.96 Gb。

。现在我如何扩展我的PGU的RAM。我想使用完整的系统内存。

Answer 1

您可以在cpu上训练以利用机器上的RAM。但是，要在gpu上运行某些东西，必须首先将其加载到gpu。现在您可以交换内存，因为并非所有结果都需要。但是，您需要花费很长的培训时间，我建议您减少批量。不过，有关此流程和实施的详细信息，请访问：https://medium.com/@Synced/how-to-train-a-very-large-and-deep-model-on-one-gpu-7b7edfe2d072。