AttributeError: module 'torch.cuda.amp' has no attribute 'initialize'

[Egg-Hu]

AttributeError: module 'torch.cuda.amp' has no attribute 'initialize'

[xmutly]

I met same error, maybe the vesion of your torch is too new.

[Wings-Of-Disaster]

You need to install APEX library