devel

makseq · makseq · commit 923925e5c1f2 · 2024-12-28T03:07:43.000Z
diff --git a/label_studio_ml/examples/grounding_dino/Dockerfile b/label_studio_ml/examples/grounding_dino/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel
 ARG DEBIAN_FRONTEND=noninteractive
 ARG TEST_ENV
 
diff --git a/label_studio_ml/examples/grounding_sam/Dockerfile b/label_studio_ml/examples/grounding_sam/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel
 ARG DEBIAN_FRONTEND=noninteractive
 ARG TEST_ENV
 
@@ -24,6 +24,7 @@ ENV PYTHONUNBUFFERED=1 \
     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"
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime`
	`1`	`+FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel`
`2`	`2`	`ARG DEBIAN_FRONTEND=noninteractive`
`3`	`3`	`ARG TEST_ENV`
`4`	`4`