diff --git a/Dockerfile.nvidia b/Dockerfile.nvidia index c972380c5..5221274c9 100644 --- a/Dockerfile.nvidia +++ b/Dockerfile.nvidia @@ -1,5 +1,5 @@ # BASE # -FROM nvidia/cuda:12.2.0-base-ubuntu20.04 AS base +FROM nvidia/cuda:12.5.1-base-ubuntu20.04 AS base WORKDIR /app ARG TARGETPLATFORM