I've been running this notebook with the Runtime Type as "high-RAM" "GPU." I was getting the following error:
CUDA out of memory. Tried to allocate 64.00 MiB (GPU 0; 15.90 GiB total capacity; 14.81 GiB already allocated; 31.75 MiB free; 14.94 GiB reserved in total by PyTorch)
So I upgraded from Pro to Pro+, because that's supposed to give me more memory, but I'm still getting the same error.