ScanSSD/utils/helpers.py

import os
import numpy as np

def get_freer_gpu():
    '''
    Find which gpu is free
    '''
    os.system('nvidia-smi -q -d Memory |grep -A4 GPU|grep Free >tmp')
    memory_available = [int(x.split()[2]) for x in open('tmp', 'r').readlines()]
    return int(np.argmax(memory_available))