From 07848144120c99abd3e9ebba2be0e04939a9c813 Mon Sep 17 00:00:00 2001 From: Yong Wu Date: Fri, 26 Apr 2024 13:20:09 -0700 Subject: [PATCH] [CI] Upgrade CUDA to 12.4 --- 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 03f34ebc70d8..acb0310a41e2 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