search

IntuitionMachines / OrigamiNet

Public implementation of our CVPR Paper "OrigamiNet: Weakly-Supervised, Segmentation-Free, One-Step, Full Page TextRecognition by learning to unfold"
142 stars 40 forks source link

Import error: No module named 'apex.parallel' #11

Open KarryBanana opened 3 years ago

KarryBanana commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

+1

jmhjimenez commented 3 years ago

You can comment it out. It worked for me.

v-nhandt21 commented 3 years ago

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