open-mmlab / mmcv

OpenMMLab Computer Vision Foundation
https://mmcv.readthedocs.io/en/latest/
Apache License 2.0
5.9k stars 1.65k forks source link

[Bug] Building wheels extremely slow on Google Colab (cu121, torch2.2.1) #3081

Open odusseys opened 7 months ago

odusseys commented 7 months ago

Prerequisite

Environment

I have tried installing from the following 2 ways:

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu122/torch2.2.1/index.html
pip install openmim
mim install mmcv-full==1.7.2

Both hang on "Building wheels for collected packages: mmcv-full" - I don't even know if it will resolve as I have not tried running it for more than 10 minutes as Colab is expensive, and requires to reinstall dependencies each time.

Is this expected?

Reproduces the problem - code sample

pip install openmim
mim install mmcv-full==1.7.2

Reproduces the problem - command or script

pip install openmim
mim install mmcv-full==1.7.2

Reproduces the problem - error message

Building wheels for collected packages: mmcv-full

Additional information

No response

zhouzaida commented 7 months ago

hi, you could try to install ninja and psutil to speed up the building process.

pip install ninja psutil
krishwin commented 1 month ago

i ran into same issue but in kaggle. ninja and psutil already installed but still mmcv-full is stuck building wheels