From 037a68833fdb402dc2d509931bc3900be94bc51c Mon Sep 17 00:00:00 2001 From: Arjun Date: Thu, 14 May 2026 22:59:36 +0000 Subject: [PATCH] Update Validator for /usr/bin symlink Signed-off-by: Arjun --- cmd/nvidia-validator/main.go | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/cmd/nvidia-validator/main.go b/cmd/nvidia-validator/main.go index 8ffa41e28..9a6f46913 100644 --- a/cmd/nvidia-validator/main.go +++ b/cmd/nvidia-validator/main.go @@ -749,7 +749,17 @@ func validateHostDriver(silent bool) error { disableDevCharSymlinkCreation = true return nil } - fileInfo, err := os.Lstat("/host/usr/bin/nvidia-smi") + + nvidiaSMIPath := "/host/usr/bin/nvidia-smi" + if target, err := os.Readlink("/host/usr/bin"); err == nil { + if filepath.IsAbs(target) { + nvidiaSMIPath = filepath.Join("/host", strings.TrimPrefix(target, "/"), "nvidia-smi") + } else { + nvidiaSMIPath = filepath.Join("/host/usr", target, "nvidia-smi") + } + } + + fileInfo, err := os.Lstat(nvidiaSMIPath) if err != nil { return fmt.Errorf("no 'nvidia-smi' file present on the host: %w", err) }