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

IDEA-Research / Grounded-SAM-2

Grounded SAM 2: Ground and Track Anything in Videos with Grounding DINO, Florence-2 and SAM 2

https://arxiv.org/abs/2401.14159

Apache License 2.0

714 stars 51 forks source link

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

Closed Masrur02 closed 1 week ago

Masrur02 commented 3 weeks ago

Hi,

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

TIA

siberianspot commented 3 weeks ago

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.

Masrur02 commented 3 weeks ago

Hi Thanks. The issue is solved.