Skip to content

Commit f896e86

Browse files
committed
Fixes in dino, sam
1 parent ccb29b2 commit f896e86

File tree

3 files changed

+0
-3
lines changed

3 files changed

+0
-3
lines changed

label_studio_ml/examples/grounding_dino/Dockerfile

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ ENV PYTHONUNBUFFERED=1 \
2424
CUDA_HOME=/usr/local/cuda \
2525
GROUNDINGDINO_REPO_PATH=/GroundingDINO
2626

27-
RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
2827
ENV CUDA_HOME=/opt/conda \
2928
TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
3029

label_studio_ml/examples/grounding_sam/Dockerfile

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ ENV PYTHONUNBUFFERED=1 \
2424
CUDA_HOME=/usr/local/cuda \
2525
GROUNDINGDINO_REPO_PATH=/GroundingDINO
2626

27-
RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
2827
ENV CUDA_HOME=/opt/conda \
2928
TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
3029

label_studio_ml/examples/yolo/Dockerfile

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@ ENV PYTHONUNBUFFERED=1 \
2323
THREADS=4 \
2424
CUDA_HOME=/usr/local/cuda
2525

26-
# RUN conda install -c "nvidia/label/cuda-12.1.1" cuda -y
2726
ENV CUDA_HOME=/opt/conda \
2827
TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
2928

0 commit comments

Comments
 (0)