Added Dockerfile along with a file to test Docker (#307)

Author: teenaxta

Dockerfile

@@ -0,0 +1,35 @@
+FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
+ARG DEBIAN_FRONTEND=noninteractive
+
+ENV CUDA_HOME=/usr/local/cuda \
+     TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX" \
+     SETUPTOOLS_USE_DISTUTILS=stdlib
+
+RUN conda update conda -y
+
+# Install libraries in the brand new image. 
+RUN apt-get -y update && apt-get install -y --no-install-recommends \
+         wget \
+         build-essential \
+         git \
+         python3-opencv \
+         ca-certificates && \
+    rm -rf /var/lib/apt/lists/*
+
+# Set the working directory for all the subsequent Dockerfile instructions.
+WORKDIR /opt/program
+
+RUN git clone https://github.com/IDEA-Research/GroundingDINO.git
+
+RUN mkdir weights ; cd weights ; wget -q https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth ; cd ..
+
+RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
+ENV CUDA_HOME=$CONDA_PREFIX
+
+ENV PATH=/usr/local/cuda/bin:$PATH
+
+RUN cd GroundingDINO/ && python -m pip install .
+
+COPY docker_test.py docker_test.py
+
+CMD [ "python", "docker_test.py" ]

docker_test.py

@@ -0,0 +1,8 @@
+from groundingdino.util.inference import load_model, load_image, predict, annotate
+import torch
+import cv2
+
+model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.pyy", "weights/groundingdino_swint_ogc.pth")
+model = model.to('cuda:0')
+print(torch.cuda.is_available())
+print('DONE!')