How to install mmcv-full with cuda11.4 and pytorch1.11.0[Docs]

open-mmlab / mmcv

OpenMMLab Computer Vision Foundation

https://mmcv.readthedocs.io/en/latest/

Apache License 2.0

5.86k stars 1.64k forks source link

How to install mmcv-full with cuda11.4 and pytorch1.11.0[Docs] #2650

Closed YaqinLong closed 1 year ago

YaqinLong commented 1 year ago

📚 The doc issue

I want to deploy a algorithm on Jetson Xavier NX, whose cuda is 11.4. In the documentation of how to install mmcv-full, it does not have the cuda11.4 option. So, how to install the mmcv-full with cuda 11.4? Thank you for your replying.

Suggest a potential alternative/fix

No response

HAOCHENYE commented 1 year ago

Hi, suggest using openmim to install mmcv-full.

pip uninstall mmcv-full -y
pip install -U openmim
mim install mmcv-full

If there is no valid pre-built package for the corresponding PyTorch and cudatoolkit, it will fallback to build mmcv from source automatically.

zhouzaida commented 1 year ago

Hi @YaqinLong , you can refer to https://mmcv.readthedocs.io/en/latest/get_started/build.html#build-mmcv-full to install it from the source code.