pipeline_parallel - ModuleNotFoundError: No module named 'amp_C'

Hi,

I was using apex and everything worked well until this week when I get the following error:

---> 39 from apex import amp
---> 20 from . import transformer
----> 3 from apex.transformer import pipeline_parallel
----> 1 from apex.transformer.pipeline_parallel.schedules import get_forward_backward_func
----> 4 from apex.transformer.pipeline_parallel.utils import get_num_microbatches
---> 23 import amp_C

ModuleNotFoundError: No module named 'amp_C'

I saw this error occurs in other scenarios (https://github.com/NVIDIA/apex/issues/865), but not at the same place. Do you know how to solve it (PS: I’m running the code in Google colab). Thanks,

Issue Analytics

State:
Created 2 years ago
Reactions:2
Comments:10

Top GitHub Comments

6reactions

lunasaracommented, Nov 2, 2021

I have the same issue. The workaround that worked for me is to use the version prior to the last commit here: https://github.com/NVIDIA/apex/tree/3303b3e7174383312a3468ef390060c26e640cb1

As you can see on the commit history the pipeline_parallels was added on October 27th here: https://github.com/NVIDIA/apex/pull/1202 I don’t know what the issue might be but I hope the woraround works for you aswell.

0reactions

PhdShicommented, Aug 19, 2022

not solved either

Top Results From Across the Web

Modulenotfounderror: No Module Named 'Globalstate' - ADocLib

Compile and run: An error occurred: import tkinter as tk # python3 ModuleNotFoundError: No module named 'tkinter'.Solution: sudo aptget install python3tk.

Cython ImportError: No module named parallel

I'm using Cython 0.15+. cython.parallel exists in Shadow.py: import sys sys.modules['cython.parallel'] = CythonDotParallel().

ModuleNotFoundError: No module named 'pipeline'

Hi, My Python program is throwing following error: ModuleNotFoundError: No module named 'pipeline' How to remove the ModuleNot.

ImportError: No module named parallel

when I "import testp" Python tells me: "ImportError: No module named parallel". And in fact, if I browse the Cython package in the...

notebookffad25d0e2

Explore and run machine learning code with Kaggle Notebooks | Using data from 60k Stack Overflow Questions with Quality Rating.