Add PyTorch GPU test script

Changes

1 changed files (+5/-0)

• gpu.py (new)
  • @@ -0,0 +1,5 @@
    + import torch + + print(torch.cuda.is_available()) + print(torch.version.hip) + cuda = torch.device('cuda')