Did you ever solve this problem? I am encountering the exact same error. I have tried restarting the kernel. tried to use torch.cuda.empty_cache()
and gc.collect()
to clear the GPU, but I’m still running into problems. I’m getting the following when running the lesson3-planet notebook directly: