From 0c2f709a4f76b052b195c5b1760d47c227cc4ee2 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Tue, 4 Apr 2023 15:43:25 -0400
Subject: [PATCH] change docker base image to older cuda

---
 tts/vits/Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tts/vits/Dockerfile b/tts/vits/Dockerfile
index c5e0089..acb5efe 100644
--- a/tts/vits/Dockerfile
+++ b/tts/vits/Dockerfile
@@ -1,6 +1,8 @@
 # ==================== BASE CONTAINER + REQUIREMENTS ====================
 
-FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+# NB: Make sure the CUDA version supports the driver version:
+# https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 
 # See: https://github.com/opencontainers/image-spec/blob/master/annotations.md
 LABEL org.opencontainers.image.title='Storyteller ML : VITS'