Add support for any number of GPUs

2022-10-22 18:09:47 -05:00 · 2022-10-22 18:09:47 -05:00 · c3de89f678
commit c3de89f678
parent c0bf1cdf2b
1 changed files with 1 additions and 2 deletions
--- a/Demos/gpu_temp_demo.py
+++ b/Demos/gpu_temp_demo.py
@ -2,8 +2,7 @@ import GPUtil
 import time

 gpus = GPUtil.getGPUs()
-gpu = gpus[0]

 while True:
-    print(str(gpu.temperature) + " C")
+    print(str(max(gpu.temperature for gpu in gpus)) + " C")
    time.sleep(3)