Data Science Asked by bjernie on April 19, 2021
I want to train a model running on tensorflow. I have a GPU but it only has 6gb of VRAM. So i was wondering if it is possible to use some of the CPU’s RAM to offload the GPU? I know it will be much slower, and i can reduce the batch size and number of layers etc. Can it be done?
If I understand correctly you are essentially looking for a way to use the CPU's RAM as a swap for the GPU's RAM. Unfortunately this isn't as easy to accomplish and might require some low level work. So if you're looking for a simple argument to add to your keras model, as far as I know there is none. Some options:
swap_memory
parameter). However these are now deprecated in favor of the equivalent keras layer, that to my knowledge does not support this functionality. Theoretically you could use compatibility mode to utilize this.Answered by Djib2011 on April 19, 2021
Get help from others!
Recent Questions
Recent Answers
© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP