diff --git a/.ci/docker/build.sh b/.ci/docker/build.sh index 42fb88997d..f40c45fea3 100755 --- a/.ci/docker/build.sh +++ b/.ci/docker/build.sh @@ -11,7 +11,7 @@ IMAGE_NAME="$1" shift export UBUNTU_VERSION="22.04" -export CUDA_VERSION="12.4.1" +export CUDA_VERSION="12.6.3" export BASE_IMAGE="nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}" echo "Building ${IMAGE_NAME} Docker image"