Fixed nvidia_smi.py
@@ -17,7 +17,7 @@ def list_gpus():
def get_ram(gpu):
try:
stdout = subprocess.check_output(
- ['nvidia-smi', '--query-gpu=memory.total', '--format=csv,noheader,nounits', '--id', str(gpu)])
+ ['nvidia-smi', '--query-gpu=memory.total', '--format=csv,noheader,nounits', '--id=%d' % gpu])
return int(stdout.strip()) * 1024 * 1024
except subprocess.CalledProcessError:
return 0
