Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added Dockerfile for CUDA compatibility, including test file #307

Merged
merged 1 commit into from
Mar 11, 2024

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

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