!pip install --upgrade torch torchvision
import torch import torchvision import torch.nn torch.cuda.is_available()
