0a3c1aa4d67cb9c49f22a87bef180c07dfafec8a,testing/get_gpu_index.py,,,#,5

Before Change

import subprocess

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:]]

After Change

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]]
if free_memory[0] > 1000:
    print("1")

Instances