Module 13.3 Google Colab GPU

# Check if GPU is available
print("GPU available:", torch.cuda.is_available())

# Print GPU name (if available)
if torch.cuda.is_available():
    print("GPU device name:", torch.cuda.get_device_name(0))