No, I wouldn’t switch back to an older version.
I think you should be fine, since the binaries ship with its own libs.
Could you check it with print(torch.version.cuda)?
What do you mean by “the for-loop is not iterating”?
Does your code just exit or does it hang in the loop?
Are you able to get a single sample from your DataLoader?
loader_iter = iter(loader)
data, target = loader_iter.next()