Python之FLAGS用法

 

from absl import app
from absl import flags

FLAGS = flags.FLAGS

flags.DEFINE_string('gpu', None, 'comma separated list of GPU(s) to use.')

def main(argv):
    del argv
    if FLAGS.gpu:
        os.environ['CUDA_VISIBLE_DEVICES'] = FLAGS.gpu
    else:
        print('Please assign GPUs.')
        exit()

if __name__ == '__main__':
    app.run(main)

 

python test.py --gpu 0