How to extend tensorflow's GPU memory from system RAM

Viewed 4170

I want to train googles object detection with faster_rcnn_with resnet101 using mscoco datasetcode. I used only 10,000 images for training purpose.I used graphics: GeForce 930M/PCIe/SSE2. NVIDIA Driver Version:384.90. here is the picture of my GeForce.enter image description here

And I have 8Gb RAM but in tensorflow gpu it is showed 1.96 Gb.

system configurationterminal show of GPU. Now How can I extend my PGU's RAM. I want to use full system memory.

1 Answers

You can train on the cpu to take advantage of the RAM on your machine. However, to run something on the gpu it has to be loaded to the gpu first. Now you can swap memory in and out, because not all the results are needed at any step. However, you pay with a very long training time and I would rather advise you to reduce the batch size. Nevertheless, details about this process and implementation can be found here: https://medium.com/@Synced/how-to-train-a-very-large-and-deep-model-on-one-gpu-7b7edfe2d072.

Related