Add HasNvidiaSmi

fffaraz · Jun 8, 2022 · b3e09cd · b3e09cd
1 parent a16ce7a
commit b3e09cd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/cmd/nvidia-smi-json.go b/cmd/nvidia-smi-json.go
@@ -14,7 +14,11 @@ func main() {
 	flag.BoolVar(&query, "q", false, "run nvidia-smi -q")
 	flag.Parse()
 	if query {
-		fmt.Println(nvidiasmijson.XmlToJson(nvidiasmijson.RunNvidiaSmi()))
+		if nvidiasmijson.HasNvidiaSmi() {
+			fmt.Println(nvidiasmijson.XmlToJson(nvidiasmijson.RunNvidiaSmi()))
+		} else {
+			fmt.Fprintln(os.Stderr, "nvidia-smi not found")
+		}
 		return
 	}
 	input, err := ioutil.ReadAll(os.Stdin)

diff --git a/nvidiasmijson.go b/nvidiasmijson.go
@@ -13,6 +13,7 @@ func XmlToObject(input []byte) *NvidiaSmiLog {
 	var data NvidiaSmiLog
 	if err := xml.Unmarshal(input, &data); nil != err {
 		fmt.Fprintln(os.Stderr, "Error unmarshalling from XML:", err)
+		return nil
 	}
 	return &data
 }
@@ -27,6 +28,11 @@ func XmlToJson(input []byte) string {
 	return string(output)
 }
 
+func HasNvidiaSmi() bool {
+	_, err := exec.LookPath("nvidia-smi")
+	return err == nil
+}
+
 func RunNvidiaSmi() []byte {
 	cmd := "nvidia-smi"
 	if runtime.GOOS == "windows" {