open-mmlab/mmyolo

Cannot install with CUDA 12.1

a-d-wicklund opened this issue · 1 comments

Prerequisite

🐞 Describe the bug

I'm unable to install mmyolo with my CUDA 12.1 installation due to the requirement for the mmcv version to be >=2.0.0rc4,<2.1.0. Any version of mmcv that is below 2.1.0 does not appear to be installable with newer pytorch versions, as outlined in this issue: open-mmlab/mmcv#2860

I think there needs to be an update to mmyolo such that mmcv==2.1.0 is accepted, because then CUDA 12.1 will be compatible.

Environment

pytorch==2.2.1+cu121

Additional information

No response