Update fn.py

fn.py

@@ -24,14 +24,22 @@ def run():
         ramg = round(ram.available / 1073741824)
 
         # vram
+        gpu = 0
+        vrama = 0
+        vramt = 0
+        vramp = 0
+        vramg = 0
+
         cmd = 'nvidia-smi --query-gpu=utilization.gpu,memory.used,memory.total --format=csv,noheader,nounits'
         csv = subprocess.run(cmd, shell=True, capture_output=True, text=True).stdout
-        r = csv.split(',')
-        gpu = round(float(r[0]))
-        vrama = round(float(r[1]))
-        vramt = round(float(r[2]))
-        vramp = round(vrama / vramt * 100)
-        vramg = round(float(r[1]) / 1024)
+        lines = csv.split("\n")
+        for line in lines:
+            r = line.split(',')
+            gpu += round(float(r[0]))
+            vrama += round(float(r[1]))
+            vramt += round(float(r[2]))
+            vramp += round(vrama / vramt * 100)
+            vramg += round(float(r[1]) / 1024)
 
         # network
         # 1秒ぶんだけ取りたかった