Set CUDA 13.1.0 as default in Dockerfile

yushangdi · claude · yushangdi · commit 182897e8f9a6 · 2026-06-03T20:24:21.000Z
The previous change to build.sh didn't trigger a Docker image rebuild
because the image hash is calculated from the Dockerfile, not build.sh.

Setting the default BASE_IMAGE in the Dockerfile ensures the hash
changes and the image is rebuilt with CUDA 13.1.0, which is required
for the cudaGraphNodeGetToolsId API used in the CUDA graph annotations
tutorial.

Co-Authored-By: Claude Sonnet 4.5 &lt;noreply@anthropic.com&gt;
diff --git a/.ci/docker/Dockerfile b/.ci/docker/Dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE
+ARG BASE_IMAGE=nvidia/cuda:13.1.0-devel-ubuntu22.04
 FROM ${BASE_IMAGE}
 
 ENV DEBIAN_FRONTEND noninteractive

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-ARG BASE_IMAGE`
	`1`	`+ARG BASE_IMAGE=nvidia/cuda:13.1.0-devel-ubuntu22.04`
`2`	`2`	`FROM ${BASE_IMAGE}`
`3`	`3`
`4`	`4`	`ENV DEBIAN_FRONTEND noninteractive`