S_M:

Hi, I am trying to run the ResNet18 model using GTX 1080 GPU but the Cuda out-of-memory error arises. I reduced the baches size to 4 and use


1080 is more than enough to handle batch size of 64 with ease on 1080. It is relatively small network.
Look for cuda memory leak somewhere else , maybe you keep referencing/copying some tensors you don’t need etc.