autogluon icon indicating copy to clipboard operation
autogluon copied to clipboard

[AutoMM] GroundingDino used in AutoMM doesn't support CUDA 12 and Torch 2.1

Open tonyhoo opened this issue 1 year ago • 1 comments

Currently we are installing groundingdino from here. However, it got issue with Torch 2.1 and CUDA 12. See log

tonyhoo avatar Nov 14 '23 20:11 tonyhoo

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 avatar Feb 01 '24 17:02 rxjx

@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.

zhiqiangdon avatar Apr 02 '24 17:04 zhiqiangdon

Fixed by #3974

zhiqiangdon avatar Apr 02 '24 17:04 zhiqiangdon