From 4017515acbc531103ef7085cdffd03996803efdd Mon Sep 17 00:00:00 2001 From: Tariq Ibrahim Date: Mon, 22 Dec 2025 11:42:56 -0800 Subject: [PATCH] add support for driver version 590.48.01 Signed-off-by: Tariq Ibrahim --- .common-ci.yml | 6 +++--- .github/workflows/image.yaml | 1 + .nvidia-ci.yml | 2 +- rhel8/install.sh | 17 +++++++++++++---- rhel9/install.sh | 17 +++++++++++++---- ubuntu22.04/install.sh | 16 ++++++++++++---- ubuntu24.04/install.sh | 16 ++++++++++++---- versions.mk | 2 +- 8 files changed, 56 insertions(+), 21 deletions(-) diff --git a/.common-ci.yml b/.common-ci.yml index 48ee7129b..9633db957 100644 --- a/.common-ci.yml +++ b/.common-ci.yml @@ -19,7 +19,7 @@ default: command: ["--experimental"] variables: - DRIVER_VERSIONS: 535.274.02 570.195.03 580.105.08 + DRIVER_VERSIONS: 535.274.02 570.195.03 580.105.08 590.48.01 BUILD_MULTI_ARCH_IMAGES: "true" stages: @@ -81,13 +81,13 @@ trigger-pipeline: .driver-versions: parallel: matrix: - - DRIVER_VERSION: [535.274.02, 570.195.03, 580.105.08] + - DRIVER_VERSION: [535.274.02, 570.195.03, 580.105.08, 590.48.01] # Define the driver versions for jobs that can be run in parallel .driver-versions-ubuntu24.04: parallel: matrix: - - DRIVER_VERSION: [570.195.03, 580.105.08] + - DRIVER_VERSION: [570.195.03, 580.105.08, 590.48.01] # Define the matrix of precompiled jobs that can be run in parallel for ubuntu22.04 .driver-versions-precompiled-ubuntu22.04: diff --git a/.github/workflows/image.yaml b/.github/workflows/image.yaml index 213b35c46..c50771cd6 100644 --- a/.github/workflows/image.yaml +++ b/.github/workflows/image.yaml @@ -31,6 +31,7 @@ jobs: - 535.274.02 - 570.195.03 - 580.105.08 + - 590.48.01 dist: - ubuntu22.04 - ubuntu24.04 diff --git a/.nvidia-ci.yml b/.nvidia-ci.yml index 01e739526..6a530a070 100644 --- a/.nvidia-ci.yml +++ b/.nvidia-ci.yml @@ -34,7 +34,7 @@ variables: # Define the public staging registry STAGING_REGISTRY: ghcr.io/nvidia STAGING_VERSION: "${CI_COMMIT_SHORT_SHA}" - PUBLISH_VERSIONS: 535.274.02 570.195.03 580.105.08 + PUBLISH_VERSIONS: 590.48.01 .image-pull-rules: # We delay the job start to allow the public pipeline to generate the required images. diff --git a/rhel8/install.sh b/rhel8/install.sh index 323737085..93f4da2ac 100755 --- a/rhel8/install.sh +++ b/rhel8/install.sh @@ -91,7 +91,9 @@ nvidia_installer () { } fabricmanager_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y nvidia-fabricmanager-${DRIVER_VERSION}-1.el8 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y nvidia-fabricmanager-${DRIVER_VERSION}-1 else dnf install -y nvidia-fabric-manager-${DRIVER_VERSION}-1 @@ -99,7 +101,9 @@ fabricmanager_install() { } nscq_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y libnvidia-nscq-${DRIVER_VERSION}-1.el8 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y libnvidia-nscq-${DRIVER_VERSION}-1 else dnf install -y libnvidia-nscq-${DRIVER_BRANCH}-${DRIVER_VERSION}-1 @@ -108,7 +112,10 @@ nscq_install() { nvsdm_install() { if [ "$TARGETARCH" = "amd64" ]; then - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y libnvsdm-${DRIVER_VERSION}-1.el8 + return 0 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y libnvsdm-${DRIVER_VERSION}-1 return 0 fi @@ -126,7 +133,9 @@ nvlink5_pkgs_install() { } imex_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y nvidia-imex-${DRIVER_VERSION}-1.el8 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y nvidia-imex-${DRIVER_VERSION}-1 elif [ "$DRIVER_BRANCH" -ge "550" ]; then dnf install -y nvidia-imex-${DRIVER_BRANCH}-${DRIVER_VERSION}-1 diff --git a/rhel9/install.sh b/rhel9/install.sh index e53b5aa37..e926f4259 100755 --- a/rhel9/install.sh +++ b/rhel9/install.sh @@ -99,7 +99,9 @@ nvidia_installer () { } fabricmanager_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y nvidia-fabricmanager-${DRIVER_VERSION}-1.el9 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y nvidia-fabricmanager-${DRIVER_VERSION}-1 else dnf install -y nvidia-fabric-manager-${DRIVER_VERSION}-1 @@ -107,7 +109,9 @@ fabricmanager_install() { } nscq_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y libnvidia-nscq-${DRIVER_VERSION}-1.el9 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y libnvidia-nscq-${DRIVER_VERSION}-1 else dnf install -y libnvidia-nscq-${DRIVER_BRANCH}-${DRIVER_VERSION}-1 @@ -117,7 +121,10 @@ nscq_install() { # libnvsdm packages are not available for arm64 nvsdm_install() { if [ "$TARGETARCH" = "amd64" ]; then - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y libnvsdm-${DRIVER_VERSION}-1.el9 + return 0 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y libnvsdm-${DRIVER_VERSION}-1 return 0 fi @@ -135,7 +142,9 @@ nvlink5_pkgs_install() { } imex_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + dnf install -y nvidia-imex-${DRIVER_VERSION}-1.el9 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then dnf install -y nvidia-imex-${DRIVER_VERSION}-1 elif [ "$DRIVER_BRANCH" -ge "550" ]; then dnf install -y nvidia-imex-${DRIVER_BRANCH}-${DRIVER_VERSION}-1 diff --git a/ubuntu22.04/install.sh b/ubuntu22.04/install.sh index 3987e8274..5ba71328d 100755 --- a/ubuntu22.04/install.sh +++ b/ubuntu22.04/install.sh @@ -53,7 +53,9 @@ repo_setup () { } fabricmanager_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends nvidia-fabricmanager=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends nvidia-fabricmanager=${DRIVER_VERSION}-1 else apt-get install -y --no-install-recommends nvidia-fabricmanager-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 @@ -61,7 +63,9 @@ fabricmanager_install() { } nscq_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends libnvidia-nscq=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends libnvidia-nscq=${DRIVER_VERSION}-1 else apt-get install -y --no-install-recommends libnvidia-nscq-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 @@ -71,7 +75,9 @@ nscq_install() { # libnvsdm packages are not available for arm64 nvsdm_install() { if [ "$TARGETARCH" = "amd64" ]; then - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends libnvsdm=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends libnvsdm=${DRIVER_VERSION}-1 elif [ "$DRIVER_BRANCH" -ge "570" ]; then apt-get install -y --no-install-recommends libnvsdm-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 @@ -86,7 +92,9 @@ nvlink5_pkgs_install() { } imex_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends nvidia-imex=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends nvidia-imex=${DRIVER_VERSION}-1 elif [ "$DRIVER_BRANCH" -ge "550" ]; then apt-get install -y --no-install-recommends nvidia-imex-${DRIVER_BRANCH}=${DRIVER_VERSION}-1; diff --git a/ubuntu24.04/install.sh b/ubuntu24.04/install.sh index fa150c679..d3669f0ac 100755 --- a/ubuntu24.04/install.sh +++ b/ubuntu24.04/install.sh @@ -45,7 +45,9 @@ setup_cuda_repo() { } fabricmanager_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends nvidia-fabricmanager=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends nvidia-fabricmanager=${DRIVER_VERSION}-1 else apt-get install -y --no-install-recommends nvidia-fabricmanager-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 @@ -53,7 +55,9 @@ fabricmanager_install() { } nscq_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends libnvidia-nscq=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends libnvidia-nscq=${DRIVER_VERSION}-1 else apt-get install -y --no-install-recommends libnvidia-nscq-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 @@ -63,7 +67,9 @@ nscq_install() { # libnvsdm packages are not available for arm64 nvsdm_install() { if [ "$TARGETARCH" = "amd64" ]; then - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends libnvsdm=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends libnvsdm=${DRIVER_VERSION}-1 elif [ "$DRIVER_BRANCH" -ge "570" ]; then apt-get install -y --no-install-recommends libnvsdm-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 @@ -78,7 +84,9 @@ nvlink5_pkgs_install() { } imex_install() { - if [ "$DRIVER_BRANCH" -ge "580" ]; then + if [ "$DRIVER_BRANCH" -ge "590" ]; then + apt-get install -y --no-install-recommends nvidia-imex=${DRIVER_VERSION}-0ubuntu1 + elif [ "$DRIVER_BRANCH" -ge "580" ]; then apt-get install -y --no-install-recommends nvidia-imex=${DRIVER_VERSION}-1 elif [ "$DRIVER_BRANCH" -ge "550" ]; then apt-get install -y --no-install-recommends nvidia-imex-${DRIVER_BRANCH}=${DRIVER_VERSION}-1; diff --git a/versions.mk b/versions.mk index e77708b70..00a68aa6d 100644 --- a/versions.mk +++ b/versions.mk @@ -13,6 +13,6 @@ # limitations under the License. # DRIVER_VERSIONS contains latest version in all active datacenter branches -DRIVER_VERSIONS ?= 535.274.02 570.195.03 580.105.08 +DRIVER_VERSIONS ?= 535.274.02 570.195.03 580.105.08 590.48.01 GOLANG_VERSION := 1.25.5