autogluon
autogluon copied to clipboard
[AutoMM] GroundingDino used in AutoMM doesn't support CUDA 12 and Torch 2.1
Currently we are installing groundingdino from here. However, it got issue with Torch 2.1 and CUDA 12. See log
I believe this can be resolved by installing the same version of CUDA as used in PyTorch. Your error log tells you that PyTorch is using 11.7 so you need to install that. I ran into the same error. Installing a matching version of CUDA via apt, adding /usr/local/cuda-11.x to the path and setting CUDA_HOME correctly via .bashrc, and then adding the correct python headers solved this for me.
@rxjx Thanks for the suggestion! This custom installation is automatic and may cause obstacles for some users with less such knowledge. We will deprecate the current groundingdino implementation and add it back through the MMDetection support later.
Fixed by #3974