Fix grounding sam

HumanSignal · Dec 28, 2024 · 3fc0c51 · 3fc0c51
1 parent 5f44ee5
commit 3fc0c51
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/label_studio_ml/examples/grounding_sam/Dockerfile b/label_studio_ml/examples/grounding_sam/Dockerfile
@@ -12,7 +12,7 @@ RUN --mount=type=cache,target="/var/cache/apt",sharing=locked \
     && apt-get install -y git \
     && apt-get install -y wget \
     && apt-get install -y g++ freeglut3-dev build-essential libx11-dev \
-    libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev libfreeimage-dev \
+    libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev libfreeimage-dev ninja-build \
     && apt-get -y install ffmpeg libsm6 libxext6 libffi-dev python3-dev python3-pip gcc
 
 ENV PYTHONUNBUFFERED=1 \
@@ -23,10 +23,8 @@ ENV PYTHONUNBUFFERED=1 \
     THREADS=4 \
     CUDA_HOME=/usr/local/cuda \
     GROUNDINGDINO_REPO_PATH=/GroundingDINO
-
-RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
-ENV CUDA_HOME=/opt/conda \
-    TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
+ENV PATH="${CUDA_HOME}/bin:${PATH}"
+ENV TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
 
 COPY requirements.txt .
 RUN --mount=type=cache,target=${PIP_CACHE_DIR},sharing=locked \