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 3ff3daa26d [CI] Upgrade CUDA to 12.4 (#16939)
3ff3daa26d is described below
commit 3ff3daa26dd8eb377cc146b28b6b639c31282bc8
Author: Yong Wu <[email protected]>
AuthorDate: Sat Apr 27 12:11:46 2024 -0700
[CI] Upgrade CUDA to 12.4 (#16939)
---
docker/Dockerfile.ci_gpu | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
index 03f34ebc70..acb0310a41 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -17,7 +17,7 @@
# CI docker GPU env
# tag: v0.60
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
COPY utils/apt-install-and-clear.sh /usr/local/bin/apt-install-and-clear
