IDEA-Research/Grounded-SAM-2

Can CUDA 12.2 be used for the Grounded-SAM-2

Masrur02 opened this issue · 2 comments

Masrur02 commented

Hi,

Is there any way to use the CUDA 12.2 for the Grounded-SAM-2?

TIA

siberianspot commented

Install PyTorch from official CUDA 12.1 index (you’ll see the warning about using CUDA Toolkit 12.2 through the build process later, but it’ll concludes that’s everything should be fine), install additional packages required for groundingdino and SAM2 (use NumPy<2.0.0), clone groundingdino and SAM2 repos, build wheels for both from the Python env where you have PyTorch installed, install wheels.

  • 👍2
Masrur02 commented

Hi
Thanks. The issue is solved.