Hi there,
I’m having a problem with my CUDA when running transfer learned networks. My code is essentially the same as can be found on the PyTorch tutorial page for transfer learning (https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html 6), and the error is as follows:

Capture.PNG865×32 2.36 KB

I used my code to successfully train/test a variety of networks before this error came up, which leads me to believe that the GPU is storing information in memory and not flushing it; any ideas on what I could do to free up this memory?
Thank you very much!
Tomaso