diff --git a/src/pkg/nvidia/nvidia.go b/src/pkg/nvidia/nvidia.go index 1866bec..9b6ba75 100644 --- a/src/pkg/nvidia/nvidia.go +++ b/src/pkg/nvidia/nvidia.go @@ -45,7 +45,14 @@ func createNullLogger() *logrus.Logger { func GenerateCDISpec() (*specs.Spec, error) { logrus.Debugf("Generating Container Device Interface for NVIDIA") - info := info.New() + var logger *logrus.Logger + if logLevel < logrus.DebugLevel { + logger = createNullLogger() + } else { + logger = logrus.StandardLogger() + } + + info := info.New(info.WithLogger(logger)) if ok, reason := info.HasDXCore(); ok { logrus.Debugf("Generating Container Device Interface for NVIDIA: Windows is unsupported: %s", reason) @@ -67,13 +74,6 @@ func GenerateCDISpec() (*specs.Spec, error) { return nil, ErrPlatformUnsupported } - var logger *logrus.Logger - if logLevel < logrus.DebugLevel { - logger = createNullLogger() - } else { - logger = logrus.StandardLogger() - } - cdi, err := nvcdi.New(nvcdi.WithInfoLib(info), nvcdi.WithLogger(logger)) if err != nil { logrus.Debugf("Generating Container Device Interface for NVIDIA: failed to create library: %s", err)