import torch
print(torch.__version__)
print(torch.cuda.is_available())  # torch.xpu is the API for Intel GPU support