Import error
See original GitHub issueHello I’m having problems to import apex to my python project. I have installed the module following the instructions.
$ pip3 install -v --no-cache-dir ./ --user
Then when I try to import amp from apex module it raises the next error:
Traceback (most recent call last): File "train.py", line 18, in <module> from apex import amp File "/home/pharaun/.local/lib/python3.6/site-packages/apex/__init__.py", line 6, in <module> from . import amp File "/home/pharaun/.local/lib/python3.6/site-packages/apex/amp/__init__.py", line 1, in <module> from .amp import init, half_function, float_function, promote_function,\ File "/home/pharaun/.local/lib/python3.6/site-packages/apex/amp/amp.py", line 5, in <module> from .frontend import * File "/home/pharaun/.local/lib/python3.6/site-packages/apex/amp/frontend.py", line 2, in <module> from ._initialize import _initialize File "/home/pharaun/.local/lib/python3.6/site-packages/apex/amp/_initialize.py", line 9, in <module> from ._process_optimizer import _process_optimizer File "/home/pharaun/.local/lib/python3.6/site-packages/apex/amp/_process_optimizer.py", line 6, in <module> from ..optimizers import FusedSGD File "/home/pharaun/.local/lib/python3.6/site-packages/apex/optimizers/__init__.py", line 2, in <module> from .fused_adam import FusedAdam File "/home/pharaun/.local/lib/python3.6/site-packages/apex/optimizers/fused_adam.py", line 3, in <module> from amp_C import multi_tensor_adam ModuleNotFoundError: No module named 'amp_C'
Did some one have the same issue? I tried to search it but I didn’t find anything.
Thanks!
Issue Analytics
- State:
- Created 4 years ago
- Comments:6
@ptrblck @mcarilli my bad. I cleaned up fused novograd/lamb before final push but forgot to do the same for adam. I’ll push another change to fix
https://github.com/NVIDIA/apex/pull/469