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