gofortargets:


b = torch.tensor(arr, dtype=torch.int64, device=‘cuda:1’)

this makes the code keep running without stopping and doesn’t show anything. But device=‘cuda:0’ works normally.


I change the device by torch.device(“cuda:1”) but it got such problem. The program stucks at creating tensor b.