diff --git a/Dockerfile b/Dockerfile index 651fec5..1d27f1f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -ARG BASE_IMAGE=nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04 +ARG BASE_IMAGE=nvidia/cuda:12.8.0-cudnn-runtime-ubuntu24.04 # hadolint ignore=DL3006 FROM ${BASE_IMAGE} LABEL org.opencontainers.image.source="https://github.com/speaches-ai/speaches"