torch version problem

[kilometer80]

I wonder which torch version Karpathy used in this amazing tutorial? Cause my torch==2.1.2 and python==3.9.12 can't convert np.uint16 into torch.long.

[peterdonnelly1]

I'm wondering too. I am using 2.4.1 and am having trouble getting ddp working.
It runs on the two GPUs just fine until the point where I introduce the loss/gradient sync code.