Hi,
How did you installed pytorch?
I’m not sure it’s an infinite loop, what you see here is that the cpu is chatting with the gpu no?
Anyway, this line is expected to take some time, few seconds maybe.
Could you run the following lines in the repl and let me know in which one it freezes:
import torch
print(torch.cuda._initialized)
torch.cuda._lazy_init()
print(torch.cuda._initialized)
a = torch.rand(10)
a = a.cuda()