enable nvidia persistence mode

microsoft · May 1, 2024 · 9dd953c · 9dd953c
1 parent 5e6a41d
commit 9dd953c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/microsoft/testsuites/gpu/gpusuite.py b/microsoft/testsuites/gpu/gpusuite.py
@@ -415,6 +415,14 @@ def _install_driver(node: Node, log_path: Path, log: Logger) -> None:
 
     __install_driver_using_sdk(node, log, log_path)
 
+    # Enabling nvidia persistence mode which is recommended by NVIDIA
+    node.execute(
+        "nvidia-persistenced --persistence-mode",
+        sudo=True,
+        expected_exit_code=0,
+        expected_exit_code_failure_message="fail to enable nvidia persistence mode",
+    )
+
 
 def _gpu_provision_check(min_pci_count: int, node: Node, log: Logger) -> None:
     lspci = node.tools[Lspci]