Set nvidia driver parameters

Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com>
nv-gha-runners · Oct 11, 2024 · 75928e2 · 75928e2
1 parent 8aad5c3
commit 75928e2
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/linux/installers/nvidia-driver.sh b/linux/installers/nvidia-driver.sh
@@ -21,3 +21,10 @@ sudo apt-get update
 sudo apt-get -y install "nvidia-driver-${NV_DRIVER_VERSION}-server-open"
 
 sudo dpkg --purge "$(dpkg -f "${KEYRING}" Package)"
+
+# Set nvidia kernel module params
+sudo sh -c "cat <<EOF > /etc/modprobe.d/nvidia.conf
+options nvidia NVreg_NvLinkDisable=1
+options nvidia NVreg_OpenRmEnableUnsupportedGpus=1
+EOF"
+sudo update-initramfs -u