oneGPU use: python wait_gpu.py [memory size you need to use] [your command] for example: python wait_gpu.py 6000 python test.py -test_arg 1