Skip to content

Commit

Permalink
Added GroundingDINO and Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
Giuseppe Quaratino committed Feb 25, 2024
1 parent aa9a594 commit 5b2eeef
Show file tree
Hide file tree
Showing 2 changed files with 34 additions and 0 deletions.
33 changes: 33 additions & 0 deletions ros/docker/ObjectDetection/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel

# Arguments to build Docker Image using CUDA
ARG USE_CUDA=0
ARG TORCH_ARCH=
ARG DEBIAN_FRONTEND=noninteractive

ENV AM_I_DOCKER True
ENV BUILD_WITH_CUDA "${USE_CUDA}"
ENV TORCH_CUDA_ARCH_LIST "${TORCH_ARCH}"
ENV CUDA_HOME /usr/local/cuda-12.1

RUN mkdir -p /home/appuser/GrouningDINO
COPY ./GroundingDINO /home/appuser/GrouningDINO/

RUN apt-get update && apt-get install --no-install-recommends wget ffmpeg \
libsm6 libxext6 git nano vim libxrender1 libxtst6 -y \
&& apt-get clean && apt-get autoremove && rm -rf /var/lib/apt/lists/*

# Enable "universe" repository
RUN sed -i 's/main$/main universe/' /etc/apt/sources.list

# Install required packages
RUN apt-get update && apt-get install --no-install-recommends -y \
libxcb-icccm4 libxcb-image0 libxcb-keysyms1 libxcb-render-util0

ENV PATH /usr/local/cuda-12.1/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/cuda-12.1/lib64:${LD_LIBRARY_PATH}

# Set environment variables for X11
ENV DISPLAY=:0

WORKDIR /home/appuser/GrouningDINO
1 change: 1 addition & 0 deletions ros/docker/ObjectDetection/GroundingDINO
Submodule GroundingDINO added at 2b62f4

0 comments on commit 5b2eeef

Please sign in to comment.