diff --git a/internal/discover/hooks.go b/internal/discover/hooks.go index 708260e1..dfe98988 100644 --- a/internal/discover/hooks.go +++ b/internal/discover/hooks.go @@ -47,6 +47,9 @@ func FindNvidiaCTK(logger *logrus.Logger, nvidiaCTKPath string) string { return nvidiaCTKPath } + if nvidiaCTKPath == "" { + nvidiaCTKPath = nvidiaCTKExecutable + } logger.Debugf("Locating NVIDIA Container Toolkit CLI as %v", nvidiaCTKPath) lookup := lookup.NewExecutableLocator(logger, "") hookPath := nvidiaCTKDefaultFilePath