open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
8.03k stars 2.58k forks source link

Support for RTX 40 series #3282

Open CauchyComplete opened 1 year ago

CauchyComplete commented 1 year ago

I would like to use mmcv-full (version 1) on RTX 4090, which forces CUDA >= 11.8. However, install fails for RTX 4090.

I know that mmcv version 2 is available for CUDA 11.8 but I need to use the function 'revert_sync_batchnorm' from 'mmcv.cnn.utils'. So I cannot use mmcv version 2.

Is there any way to use mmcv-full with RTX 4090?

corydambach commented 6 months ago

Have you had any success? I have tried to get it working but have had no luck.