Commit 1b27f431 authored by cmaffeo2's avatar cmaffeo2
Browse files

Fixed run.py

parent 12a7248f
......@@ -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)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment