Skip to content

Conversation

@teenaxta
Copy link
Contributor

@teenaxta teenaxta commented Mar 8, 2024

Set cuda architecture to ensure that docker file compiles with CUDA on the latest pytorch docker image

@anas-zafar
Copy link

@teenaxta could you please share instructions how to run this model in docker? Thanks

@snenyl
Copy link

snenyl commented Oct 10, 2024

For people that run RTX 3060.

Dockerfile.txt

Build with the following command

docker build -t groundingdino:dev .

Run and mount with selected <host-image-folder>
docker run --gpus all -it --rm -v <host-image-folder>:/GroundingDINO/images groundingdino:dev /bin/bash

While inside the docker container run the following command

CUDA_VISIBLE_DEVICES=0 python3 demo/inference_on_a_image.py \
    --config_file groundingdino/config/GroundingDINO_SwinT_OGC.py \
    --checkpoint_path weights/groundingdino_swinb_cogcoor.pth \
    --image_path /GroundingDINO/images/store.webp \
    --text_prompt "plastic" \
    --output_dir /GroundingDINO/images/output \
    --box_threshold 0.35 \
    --text_threshold 0.25

BobbyPeng123 pushed a commit to BobbyPeng123/GroundingDINO that referenced this pull request Sep 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants