0
点赞
收藏
分享

微信扫一扫

Python之FLAGS用法

金牛豆豆 2022-07-18 阅读 65

 

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

黄世宇/Shiyu Huang's Personal Page:​​https://huangshiyu13.github.io/​​



举报

相关推荐

0 条评论