Python3(进阶)-FLAG的用法

【例】helloFLAG.py

from absl import app
from absl import flags
import os

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 helloFLAG.py --gpu 0