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

Before Change

else:
    print("0")

if os.path.exists("./gpu_tmp_log"):
    os.remove("./gpu_tmp_log")

After Change

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