diff --git a/services/nvidia/Dockerfile b/services/nvidia/Dockerfile index 8a3fc51..f8a938f 100644 --- a/services/nvidia/Dockerfile +++ b/services/nvidia/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:13.0.2-cudnn-devel-ubuntu24.04 +FROM nvidia/cuda:13.1.0-cudnn-devel-ubuntu24.04 ENV PIP_BREAK_SYSTEM_PACKAGES=1 ENV DEBIAN_FRONTEND=noninteractive