From 1d9c761ed471f3f9f4e68b3bc6c1933ba9ce3d48 Mon Sep 17 00:00:00 2001
From: daven-que <daven-que@github.com>
Date: Tue, 24 Feb 2026 19:32:19 -0500
Subject: [PATCH] Fix lspci and setpci in air-gapped environments. 
 https://github.com/NVIDIA/gpu-driver-container/issues/616

---
 vgpu-manager/rhel8/ocp_dtk_entrypoint | 9 +++++++++
 vgpu-manager/rhel9/ocp_dtk_entrypoint | 9 +++++++++
 2 files changed, 18 insertions(+)

diff --git a/vgpu-manager/rhel8/ocp_dtk_entrypoint b/vgpu-manager/rhel8/ocp_dtk_entrypoint
index cf3627ba1..519b1b3e9 100755
--- a/vgpu-manager/rhel8/ocp_dtk_entrypoint
+++ b/vgpu-manager/rhel8/ocp_dtk_entrypoint
@@ -20,6 +20,8 @@ nv-ctr-run-with-dtk() {
            /usr/local/bin/ocp_dtk_entrypoint \
            /usr/local/bin/nvidia-driver \
            /usr/sbin/lspci \
+           /usr/sbin/setpci \
+           /lib64/libpci.so.* \
            /driver \
            "$DRIVER_TOOLKIT_SHARED_DIR/"
 
@@ -96,10 +98,17 @@ dtk-build-driver() {
     cp -v \
        "$DRIVER_TOOLKIT_SHARED_DIR/nvidia-driver" \
        "$DRIVER_TOOLKIT_SHARED_DIR/lspci" \
+       "$DRIVER_TOOLKIT_SHARED_DIR/setpci" \
        "${DRIVER_TOOLKIT_SHARED_DIR}/bin"
 
     export PATH="${DRIVER_TOOLKIT_SHARED_DIR}/bin:$PATH";
 
+    mkdir "${DRIVER_TOOLKIT_SHARED_DIR}/lib" -p
+    cp -v \
+       "$DRIVER_TOOLKIT_SHARED_DIR"/libpci.so.* \
+       "${DRIVER_TOOLKIT_SHARED_DIR}/lib"
+    export LD_LIBRARY_PATH="${DRIVER_TOOLKIT_SHARED_DIR}/lib:$LD_LIBRARY_PATH";
+
     # ensure lspci is installed, as 'sriov-manage' script requires it
     if ! $(lspci >/dev/null); then
       dnf install -y pciutils && rm -rf /var/cache/yum/*
diff --git a/vgpu-manager/rhel9/ocp_dtk_entrypoint b/vgpu-manager/rhel9/ocp_dtk_entrypoint
index 3e3040317..28f82de01 100755
--- a/vgpu-manager/rhel9/ocp_dtk_entrypoint
+++ b/vgpu-manager/rhel9/ocp_dtk_entrypoint
@@ -32,6 +32,8 @@ nv-ctr-run-with-dtk() {
            /usr/local/bin/ocp_dtk_entrypoint \
            /usr/local/bin/nvidia-driver \
            /usr/sbin/lspci \
+           /usr/sbin/setpci \
+           /lib64/libpci.so.* \
            /driver \
            "$DRIVER_TOOLKIT_SHARED_DIR/"
 
@@ -108,10 +110,17 @@ dtk-build-driver() {
     cp -v \
        "$DRIVER_TOOLKIT_SHARED_DIR/nvidia-driver" \
        "$DRIVER_TOOLKIT_SHARED_DIR/lspci" \
+       "$DRIVER_TOOLKIT_SHARED_DIR/setpci" \
        "${DRIVER_TOOLKIT_SHARED_DIR}/bin"
 
     export PATH="${DRIVER_TOOLKIT_SHARED_DIR}/bin:$PATH";
 
+    mkdir "${DRIVER_TOOLKIT_SHARED_DIR}/lib" -p
+    cp -v \
+       "$DRIVER_TOOLKIT_SHARED_DIR"/libpci.so.* \
+       "${DRIVER_TOOLKIT_SHARED_DIR}/lib"
+    export LD_LIBRARY_PATH="${DRIVER_TOOLKIT_SHARED_DIR}/lib:$LD_LIBRARY_PATH";
+
     # ensure lspci is installed, as 'sriov-manage' script requires it
     if ! $(lspci >/dev/null); then
       dnf install -y pciutils && rm -rf /var/cache/yum/*