diff --git a/cmd/nvidia-mig-manager/main.go b/cmd/nvidia-mig-manager/main.go index 2fc861e8..90e426c4 100644 --- a/cmd/nvidia-mig-manager/main.go +++ b/cmd/nvidia-mig-manager/main.go @@ -47,7 +47,7 @@ const ( DefaultHostMigManagerStateFile = "/etc/systemd/system/nvidia-mig-manager.service.d/override.conf" DefaultHostKubeletSystemdService = "kubelet.service" DefaultGPUClientsNamespace = "default" - DefaultDriverRoot = "/run/nvidia/driver" + DefaultNvidiaDriverRoot = "/run/nvidia/driver" DefaultDriverRootCtrPath = "/run/nvidia/driver" ) @@ -211,11 +211,10 @@ func main() { }, &cli.StringFlag{ Name: "driver-root", - Aliases: []string{"t"}, - Value: DefaultDriverRoot, - Usage: "Root path to the NVIDIA driver installation. Only used if --cdi-enabled is set.", + Aliases: []string{"nvidia-driver-root", "t"}, + Value: DefaultNvidiaDriverRoot, Destination: &driverRoot, - EnvVars: []string{"DRIVER_ROOT"}, + EnvVars: []string{"NVIDIA_DRIVER_ROOT", "DRIVER_ROOT"}, }, &cli.StringFlag{ Name: "driver-root-ctr-path",