Add PyTorch GPU test script

This commit is contained in:
Anthony Wang 2021-12-14 10:11:33 -06:00
parent 55e855ad86
commit 6be866b0b8

5
gpu.py Normal file
View File

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