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

After Change

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")
else:

Instances