ShoufaChen / DiffusionDet

[ICCV2023 Best Paper Finalist] PyTorch implementation of DiffusionDet (https://arxiv.org/abs/2211.09788)
Other
2.08k stars 163 forks source link

Detectron's version of cuda #102

Open Yu1802 opened 11 months ago

Yu1802 commented 11 months ago

I want to run vitdet model with recent mmdet, but need to install detectron2 first. According to mmdet environment, I configured python=3.8, cuda=12.1 configuration. But detectron2 currently only supports up to cuda11.3. I would like to ask if detectron supports higher versions or if it is feasible to lower the torch version? Thank you.