I have a problem with the memory of my GPU, when I try to use the functions lr_find() and fit_one_cycle()
RuntimeError: CUDA out of memory. Tried to allocate 1024.00 MiB (GPU 0; 10.75 GiB total capacity; 9.16 GiB already allocated; 79.06 MiB free; 9.79 GiB reserved in total by PyTorch)
I am using images of 1024 x 1024 and a GeForce RTX 2080 Ti and fastai 2.3.0, I tried to do it with different batch size (128,64,32,16,8,4) even with batch size 1 and I keep having the same problem.
I already tried with these 3 solutions
kill -9 PID
I tried with CUDA 11 and CUDA 10.2
(I also restart the kernel frequently)
and the result is the same. Please if anyone has experience with this, please help me.