From 57a751521aaa44e63a6630d88ecdf7bbdd274d0b Mon Sep 17 00:00:00 2001
From: Scott Davidson <scott@stackhpc.com>
Date: Thu, 26 Oct 2023 15:27:52 +0100
Subject: [PATCH] Use cuda-runtime base instead of cuda-devel

---
 images/api-base/Dockerfile | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/images/api-base/Dockerfile b/images/api-base/Dockerfile
index 15d7211..5845599 100644
--- 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/*