Specifiy DRIVER_ROOT consistently

This change ensures that the nvidia-mig-manager CLI requires the path to the driver root accept both the NVIDIA_DRIVER_ROOT and DRIVER_ROOT environment variables in addition to the --driver-root command line argument. Signed-off-by: Christopher Desiniotis <[email protected]>
NVIDIA · Jun 15, 2024 · f4ff091 · f4ff091
1 parent d6bb24f
commit f4ff091
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmd/nvidia-mig-manager/main.go b/cmd/nvidia-mig-manager/main.go
@@ -217,7 +217,7 @@ func main() {
 			Value:       DefaultDriverRoot,
 			Usage:       "Root path to the NVIDIA driver installation. Only used if --cdi-enabled is set.",
 			Destination: &driverRoot,
-			EnvVars:     []string{"DRIVER_ROOT"},
+			EnvVars:     []string{"NVIDIA_DRIVER_ROOT", "DRIVER_ROOT"},
 		},
 		&cli.StringFlag{
 			Name:        "driver-root-ctr-path",