Hi guys,
I just upgraded my pytorch version to use AMP, and while Gradscaler now imports from torch.cuda.amp, I get the following error for autocast:
ImportError: cannot import name 'autocast' from 'torch.cuda.amp' (/home/benjamin/anaconda3/lib/python3.7/site-packages/torch/cuda/amp/__init__.py)

any idea why ?