enter image description here
Here, "model = model.cuda()"
You need to choose gpu.
from Runtime -> change runtime type and select GPU as accelerator
