Colab GPU 설정

!pip install --upgrade torch torchvision

import torch
import torchvision
import torch.nn

torch.cuda.is_available()