From 1fd33e76b47b1f4285485ce60835b01d433fc9ca Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Thu, 27 Apr 2023 01:02:23 -0700
Subject: [PATCH] update base image

---
 service_gpu.Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/service_gpu.Dockerfile b/service_gpu.Dockerfile
index 5f9253f8a5..38ff6eb8ab 100644
--- a/service_gpu.Dockerfile
+++ b/service_gpu.Dockerfile
@@ -131,7 +131,8 @@ RUN du -hsc * | sort -hr
 # Final image
 #FROM ubuntu:jammy as final
 # TODO(bt,2023-04-26): This is only necessary for download-job and inference-job
-FROM nvidia/cuda:12.0.1-runtime-ubuntu22.04 as final
+#FROM nvidia/cuda:12.0.1-runtime-ubuntu22.04 as final # Same image as so-vits-svc
+FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04 as final
 
 # See: https://github.com/opencontainers/image-spec/blob/master/annotations.md
 LABEL org.opencontainers.image.title='Storyteller Rust (GPU)'