Fixed Dockerfile to use CUDA 11.6 when compiling GroundingDINO

IDEA-Research / Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything

https://arxiv.org/abs/2401.14159

Apache License 2.0

14.88k stars 1.38k forks source link

Fixed Dockerfile to use CUDA 11.6 when compiling GroundingDINO #435

Closed matejur closed 8 months ago

matejur commented 8 months ago

When installing GroundingDino with build isolation PyTorch with CUDA 12.1 is installed.

This results in the following error:

The detected CUDA version (11.6) mismatches the version that was used to compile PyTorch (12.1). Please make sure to use the same CUDA versions.

Installing wheel and disabling build isolation correctly compiles GroundingDINO with CUDA 11.6.

rentainhe commented 8 months ago

Thanks so much for updating this!