I had my poor Nvidia 3060 Laptop GPU.
But when I tried to run the code, it simply tells me out of memory. (OutOfMemoryError: CUDA out of memory.)
So I was forced to change the code to use CPU, device = torch.device('cpu')
And it is so slow, which is annoying.
Do anyone know the minimum requirement for the GPU?
Thanks!