ModuleNotFoundError: No module named 'mxnet.contrib.amp'

[Mukulareddy]

Description

(A clear and concise description of what the bug is.)

Error Message

(Paste the complete error message. Please also include stack trace by setting environment variable DMLC_LOG_STACK_TRACE_DEPTH=10 before running your script.)

To Reproduce

(If you developed your own code, please provide a short script that reproduces the error. For existing examples, please provide link.)

Steps to reproduce

(Paste the commands you ran that produced the error.)

What have you tried to solve it?

Environment

We recommend using our script for collecting the diagnositc information. Run the following command and paste the outputs below:

curl --retry 10 -s https://raw.githubusercontent.com/dmlc/gluon-nlp/master/tools/diagnose.py | python

# paste outputs here

[ptrendx]

What is the version of MXNet you are using?

[Dragas]

Seems to be an issue when running an mxnet/python:1.4.0_cpu_py3 docker image.

[ptrendx]

AMP was introduced in MXNet 1.5. Could you try that version (or newer)?

[Radiushina]

I have the same problem - ModuleNotFoundError: No module named 'mxnet.contrib.amp'. Version MXNet is 2.0.0.

[ptrendx]

In 2.0 AMP is not in contrib anymore - you should use mxnet.amp instead.