[CI][CUDA] Uplift docker to use cuda 12.5 image. (#14049)

This upgrades the docker to use the cuda 12.5 image. I've ran the test-e2e locally using cuda 12.5 and all is well. cuda 12.5 also fixed an issue introduced by the cuda 12.4 driver: see #13661 (comment) Signed-off-by: JackAKirk <[email protected]>
intel · Jun 5, 2024 · f665968 · f665968
1 parent 0cec128
commit f665968
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/devops/containers/ubuntu2204_build.Dockerfile b/devops/containers/ubuntu2204_build.Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+FROM nvidia/cuda:12.5.0-devel-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive