aa495c2ace9444aa5028f699ae62dddef45ab7d9,testing/get_gpu_index.py,,,#,5

Before Change

rc = subprocess.call("/home/wenqi/list_gpu_memory.sh", shell=True)
gpu_info = []
with open("./gpu_tmp_log", "r") as f:
    for line in f.readlines():
        gpu_info.append(line[:-1])
free_memory = [float(x) for x in gpu_info[1:]]
if free_memory[0] > 1000:
    print("1")
else:

After Change

import os
from subprocess import Popen, PIPE

bash_string = "nvidia-smi --query-gpu=memory.free --format=csv"
p1 = Popen(bash_string.split(), stdout=PIPE)
bash_string = "tail -n 2"
p2 = Popen(bash_string.split(), stdin=p1.stdout, stdout=PIPE)
p3 = Popen(["sed", "s: MiB::"], stdin=p2.stdout, stdout=PIPE)
output, error = p3.communicate()

free_memory = [float(x) for x in output.split("\n")[:-1]]

Instances