【转】Python之FLAGS用法

最新推荐文章于 2025-07-15 23:03:29 发布

我是杨老师-AI

最新推荐文章于 2025-07-15 23:03:29 发布

阅读量2.1w

点赞数

本文介绍了一个简单的 Python 脚本，用于指定要使用的 GPU，并在运行时打印所选的 GPU。此脚本使用 absl 库进行参数解析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【转】https://www.cnblogs.com/huangshiyu13/p/8415665.html

flag_test.py

测试：python flag_test.py --gpu=abc

import sys, os
from absl import app
from absl import flags


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

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

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