From 8967507f00dc9575e5fb6e48210db17b5bffbbf1 Mon Sep 17 00:00:00 2001 From: Alexey Rivkin Date: Thu, 13 Nov 2025 14:27:26 +0200 Subject: [PATCH] CI: CUDA upgrade for AWS EFA tests Signed-off-by: Alexey Rivkin --- contrib/aws-efa/README.md | 2 +- contrib/aws-efa/aws_job_def.json | 2 +- contrib/aws-efa/aws_test.sh | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/contrib/aws-efa/README.md b/contrib/aws-efa/README.md index cd74a7e06..58e171cf5 100644 --- a/contrib/aws-efa/README.md +++ b/contrib/aws-efa/README.md @@ -89,7 +89,7 @@ The AWS test script: ## Container Image -The script uses the container image: `nvcr.io/nvidia/cuda-dl-base:25.06-cuda12.9-devel-ubuntu24.04` +The script uses the container image: `nvcr.io/nvidia/cuda-dl-base:25.10-cuda13.0-devel-ubuntu24.04` You can override this by setting the `CONTAINER_IMAGE` environment variable: ```bash diff --git a/contrib/aws-efa/aws_job_def.json b/contrib/aws-efa/aws_job_def.json index 6477a16aa..5a539fb91 100644 --- a/contrib/aws-efa/aws_job_def.json +++ b/contrib/aws-efa/aws_job_def.json @@ -15,7 +15,7 @@ "imagePullSecrets": [], "containers": [ { - "image": "nvcr.io/nvidia/cuda-dl-base:25.06-cuda12.9-devel-ubuntu24.04", + "image": "nvcr.io/nvidia/cuda-dl-base:25.10-cuda13.0-devel-ubuntu24.04", "command": [ "/bin/bash", "-c", diff --git a/contrib/aws-efa/aws_test.sh b/contrib/aws-efa/aws_test.sh index 0477757a7..44694e08f 100755 --- a/contrib/aws-efa/aws_test.sh +++ b/contrib/aws-efa/aws_test.sh @@ -30,7 +30,7 @@ usage() { echo " GITHUB_REPOSITORY - GitHub repository (e.g., \"ai-dynamo/nixl\")" echo "" echo "Optional environment variables:" - echo " CONTAINER_IMAGE - Container image to use (default: nvcr.io/nvidia/cuda-dl-base:25.06-cuda12.9-devel-ubuntu24.04)" + echo " CONTAINER_IMAGE - Container image to use (default: nvcr.io/nvidia/cuda-dl-base:25.10-cuda13.0-devel-ubuntu24.04)" echo " TEST_TIMEOUT - Timeout for test execution in minutes" exit 1 } @@ -47,7 +47,7 @@ if [ -z "$GITHUB_REF" ] || [ -z "$GITHUB_SERVER_URL" ] || [ -z "$GITHUB_REPOSITO fi test_cmd="$1" -export CONTAINER_IMAGE=${CONTAINER_IMAGE:-"nvcr.io/nvidia/cuda-dl-base:25.06-cuda12.9-devel-ubuntu24.04"} +export CONTAINER_IMAGE=${CONTAINER_IMAGE:-"nvcr.io/nvidia/cuda-dl-base:25.10-cuda13.0-devel-ubuntu24.04"} # Set Git checkout command based on GITHUB_REF case "$GITHUB_REF" in