Import error: No module named 'apex.parallel'

[KarryBanana]

I install all the dependencies with ' pip install -r requirements.txt', written in README.md.
But when I try to run the training script 'python train.py --gin iam/iam.gin',
an import error occurs: Import error: No module named 'apex.parallel'
I think I need to install NVIDIA/apex, but after I install the NVIDIA/apex, the problems still remain.
What else should I install or do I install NVIDIA/apex incorrectly ?

[v-nhandt21]

I install all the dependencies with ' pip install -r requirements.txt', written in README.md.
But when I try to run the training script 'python train.py --gin iam/iam.gin',
an import error occurs: Import error: No module named 'apex.parallel'
I think I need to install NVIDIA/apex, but after I install the NVIDIA/apex, the problems still remain.
What else should I install or do I install NVIDIA/apex incorrectly ?

I have a same problem with you, Have you solve it

[sajalroychowdhury]

+1

[jmhjimenez]

You can comment it out. It worked for me.

[v-nhandt21]

As I remember I have solved this problem by install apex from the source and try to use the correct version set of cuda, cudnn, pytorch. Check your env, the compatible can be found easily. A note that if you use CUDA11. you much install it from pytorch source.

One more thing, Apex is not used commonly these day due to there some another one is better such as deepspeed of Microsoft