Set NVIDIA_DRIVER_CAPABILITIES to all when GPU is enabled

kubernetes · Aug 20, 2024 · 2957f96 · 2957f96
1 parent 8d17e2c
commit 2957f96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/drivers/kic/oci/oci.go b/pkg/drivers/kic/oci/oci.go
@@ -191,7 +191,7 @@ func CreateContainerNode(p CreateParams) error { //nolint to suppress cyclomatic
 		runArgs = append(runArgs, "--ip", p.IP)
 	}
 	if p.GPUs != "" {
-		runArgs = append(runArgs, "--gpus", "all")
+		runArgs = append(runArgs, "--gpus", "all", "--env", "NVIDIA_DRIVER_CAPABILITIES=all")
 	}
 
 	memcgSwap := hasMemorySwapCgroup()