diff --git a/.common-ci.yml b/.common-ci.yml index 48ee7129..9633db95 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 213b35c4..c50771cd 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 01e73952..6a530a07 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 32373708..93f4da2a 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 e53b5aa3..e926f425 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 3987e827..5ba71328 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 fa150c67..d3669f0a 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 e77708b7..00a68aa6 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