add support for driver version 590.48.01

Signed-off-by: Tariq Ibrahim <[email protected]>

Author: tariq1890

.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:

.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

.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.

ubuntu22.04/install.sh

@@ -53,15 +53,19 @@ 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
   fi
 }
 
 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;

ubuntu24.04/install.sh

@@ -45,15 +45,19 @@ 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
   fi
 }
 
 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;

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