preprocessing/import torch.py

import torch

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print("Using device:", device)