blob: d8a815bb44327ed0b1981c8c41ccb6ea3e89bc2f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
// SPDX-License-Identifier: GPL-3.0-or-later
package nvidia_smi
import (
"fmt"
"os"
"os/exec"
)
func (nv *NvidiaSMI) initNvidiaSMIExec() (nvidiaSMI, error) {
binPath := nv.BinaryPath
if _, err := os.Stat(binPath); os.IsNotExist(err) {
path, err := exec.LookPath(nv.binName)
if err != nil {
return nil, fmt.Errorf("error on lookup '%s': %v", nv.binName, err)
}
binPath = path
}
return newNvidiaSMIExec(binPath, nv.Config, nv.Logger)
}
|