-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Giuseppe Quaratino
committed
Feb 25, 2024
1 parent
aa9a594
commit 5b2eeef
Showing
2 changed files
with
34 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
Submodule GroundingDINO
added at
2b62f4