import torch

avail=torch.cuda.is_available()
print(f'Is torch running on GPU?: {avail}')