fix: SAM2 image and video cuda dependencies in dockerfile

HumanSignal · Jan 8, 2025 · 08f8c4a · 08f8c4a
1 parent c2a57d5
commit 08f8c4a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/label_studio_ml/examples/segment_anything_2_image/Dockerfile b/label_studio_ml/examples/segment_anything_2_image/Dockerfile
@@ -24,7 +24,9 @@ ENV PYTHONUNBUFFERED=1 \
     CUDA_HOME=/usr/local/cuda \
     SEGMENT_ANYTHING_2_REPO_PATH=/segment-anything-2
 
-RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
+# this image is already built with cuda 12.1.1, so we don't need to install it again
+# 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"
 

diff --git a/label_studio_ml/examples/segment_anything_2_video/Dockerfile b/label_studio_ml/examples/segment_anything_2_video/Dockerfile
@@ -24,7 +24,9 @@ ENV PYTHONUNBUFFERED=1 \
     CUDA_HOME=/usr/local/cuda \
     SEGMENT_ANYTHING_2_REPO_PATH=/segment-anything-2
 
-RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
+# this image is already built with cuda 12.1.1, so we don't need to install it again
+# 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"