Fixed run.py

run.py

@@ -222,7 +222,7 @@ def run(f):
     sys.stderr = oldErr
 
 pynvml.nvmlInit()
-gpus = range(len(pynvml.nvmlDeviceGetCount()))
+gpus = range(pynvml.nvmlDeviceGetCount())
 pynvml.nvmlShutdown()
 # gpus = [0,1,2]
 print(gpus)