storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-29 09:18:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

change docker base image to older cuda

Browse Source
This commit is contained in:
Brandon Thomas
2023-04-04 15:43:25 -04:00
parent 35772ff318
commit 0c2f709a4f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/vits/Dockerfile
+3 -1
View File
@@ -1,6 +1,8 @@
# ==================== BASE CONTAINER + REQUIREMENTS ==================== # ==================== 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 # See: https://github.com/opencontainers/image-spec/blob/master/annotations.md
LABEL org.opencontainers.image.title='Storyteller ML : VITS' LABEL org.opencontainers.image.title='Storyteller ML : VITS'