No module named 'torch.cuda.amp.autocast'

[ptrblck]

Answered here.
Please try to avoid double posts.