use the nvidia-driver:XXX-dkms module stream to install FM and NSCQ pkgs

Signed-off-by: Tariq Ibrahim <tibrahim@nvidia.com>
NVIDIA · Oct 31, 2024 · 3e34c94 · 3e34c94
1 parent b44c5f8
commit 3e34c94
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rhel8/Dockerfile b/rhel8/Dockerfile
@@ -76,7 +76,7 @@ RUN if [ "$DRIVER_TYPE" != "vgpu" ] && [ "$TARGETARCH" != "arm64" ]; then \
       fmPackage=nvidia-fabricmanager-${DRIVER_BRANCH}-${DRIVER_VERSION}-1; \
     fi; \
     nscqPackage=libnvidia-nscq-${DRIVER_BRANCH}-${DRIVER_VERSION}-1; \
-    dnf module enable -y nvidia-driver:${DRIVER_BRANCH} && \
+    dnf module enable -y nvidia-driver:${DRIVER_BRANCH}-dkms && \
     dnf install -y ${fmPackage} ${nscqPackage}; fi
 
 COPY nvidia-driver /usr/local/bin

diff --git a/rhel9/Dockerfile b/rhel9/Dockerfile
@@ -70,7 +70,7 @@ RUN if [ "$DRIVER_TYPE" != "vgpu" ]; then \
 RUN if [ "$DRIVER_TYPE" != "vgpu" ] && [ "$TARGETARCH" != "arm64" ]; then \
     versionArray=(${DRIVER_VERSION//./ }); \
     DRIVER_BRANCH=${versionArray[0]}; \
-    dnf module enable -y nvidia-driver:${DRIVER_BRANCH} && \
+    dnf module enable -y nvidia-driver:${DRIVER_BRANCH}-dkms && \
     dnf install -y nvidia-fabric-manager-${DRIVER_VERSION}-1 libnvidia-nscq-${DRIVER_BRANCH}-${DRIVER_VERSION}-1; fi
 
 COPY nvidia-driver /usr/local/bin