Merge pull request #707 from elezar/align-container-driver-root-args

Align container driver root envvars
NVIDIA · May 16, 2024 · 90e08b7 · 90e08b7
2 parents 4720121 + 21d2477
commit 90e08b7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cmd/nvidia-device-plugin/main.go b/cmd/nvidia-device-plugin/main.go
@@ -115,10 +115,11 @@ func main() {
 			EnvVars: []string{"NVIDIA_CTK_PATH"},
 		},
 		&cli.StringFlag{
-			Name:    "container-driver-root",
+			Name:    "driver-root-ctr-path",
+			Aliases: []string{"container-driver-root"},
 			Value:   spec.DefaultContainerDriverRoot,
 			Usage:   "the path where the NVIDIA driver root is mounted in the container; used for generating CDI specifications",
-			EnvVars: []string{"CONTAINER_DRIVER_ROOT"},
+			EnvVars: []string{"DRIVER_ROOT_CTR_PATH", "CONTAINER_DRIVER_ROOT"},
 		},
 		&cli.StringFlag{
 			Name:    "mps-root",