Skip to content

Commit 2635f8b

Browse files
authored
fix: SAM2 image and video cuda dependencies in dockerfile (#693)
1 parent c2a57d5 commit 2635f8b

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

label_studio_ml/examples/segment_anything_2_image/Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@ ENV PYTHONUNBUFFERED=1 \
2424
CUDA_HOME=/usr/local/cuda \
2525
SEGMENT_ANYTHING_2_REPO_PATH=/segment-anything-2
2626

27-
RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
27+
# this image is already built with cuda 12.1.1, so we don't need to install it again
28+
# RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
29+
2830
ENV CUDA_HOME=/opt/conda \
2931
TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
3032

label_studio_ml/examples/segment_anything_2_video/Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@ ENV PYTHONUNBUFFERED=1 \
2424
CUDA_HOME=/usr/local/cuda \
2525
SEGMENT_ANYTHING_2_REPO_PATH=/segment-anything-2
2626

27-
RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
27+
# this image is already built with cuda 12.1.1, so we don't need to install it again
28+
# RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
29+
2830
ENV CUDA_HOME=/opt/conda \
2931
TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
3032

0 commit comments

Comments
 (0)