This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new a100bd3771 [CI] Align cuda-python with PyTorch cuda-bindings (#19738)
a100bd3771 is described below
commit a100bd3771e9aca14deea7979457726170e4b7ba
Author: Shushi Hong <[email protected]>
AuthorDate: Fri Jun 12 08:32:34 2026 -0400
[CI] Align cuda-python with PyTorch cuda-bindings (#19738)
This updates the CI GPU Docker dependency pin for `cuda-python` from
`12.8.0` to `12.9.4`.
`torch==2.10.0`, installed later by `ubuntu_install_onnx.sh`, requires
`cuda-bindings==12.9.4` on Linux x86_64. However, `cuda-python==12.8.0`
constrains `cuda-bindings` to the `12.8.x` series, leaving the CI Python
CUDA dependency set inconsistent.
Using `cuda-python==12.9.4` keeps the `cuda-bindings` requirement
aligned with the PyTorch wheel already used in the CI image. This does
not change the CUDA base image.
---
docker/install/ubuntu_install_cuda_python.sh | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/docker/install/ubuntu_install_cuda_python.sh
b/docker/install/ubuntu_install_cuda_python.sh
index dbb97f52f3..d6138a7bf2 100644
--- a/docker/install/ubuntu_install_cuda_python.sh
+++ b/docker/install/ubuntu_install_cuda_python.sh
@@ -20,4 +20,4 @@ set -e
set -u
set -o pipefail
-uv pip install cuda-python==12.8.0
+uv pip install cuda-python==12.9.4
