Updated nvidia base image

flyteorg · Oct 9, 2023 · 463a584 · 463a584
1 parent af690b6
commit 463a584
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/sandbox-bundled/Dockerfile.gpu b/docker/sandbox-bundled/Dockerfile.gpu
@@ -26,7 +26,7 @@ RUN --mount=type=cache,target=/root/.cache/go-build --mount=type=cache,target=/r
 # syntax=docker/dockerfile:1.4-labs
 
 #Following 
-FROM nvidia/cuda:11.4.0-base-ubuntu20.04
+FROM nvidia/cuda:12.1.1-base-ubuntu20.04
 
 RUN echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections