Module 'torch' has no attribute 'amp'

Hey, I’ve been having trouble to see what’s wrong here. I have torch 1.8.1+cu111

I believe the torch.amp namespace was added in PyTorch 1.12.0+ after mixed-precision training was implemented for the CPU.
In older versions, you would need to use torch.cuda.amp instead.

Thank you! The issue was resolved