Use cuda-runtime base instead of cuda-devel

stackhpc · Oct 26, 2023 · 57a7515 · 57a7515
1 parent 28bff5f
commit 57a7515
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/images/api-base/Dockerfile b/images/api-base/Dockerfile
@@ -1,7 +1,4 @@
-
-# FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
-
-FROM nvidia/cuda:12.2.2-devel-ubuntu22.04
+FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
 RUN apt-get update && \
     apt-get install -y python3-pip python3-dev && \
     rm -rf /var/lib/apt/lists/*