The recommended command-line parsing module in the Python standard library.
argparse = argparse.ArgumentParser()
argparse.add_argument('-G','--gpus',type=str,help='using gpus',default='7')
args = argparse.parse_args()
os.environ["CUDA_VISIBLE_DEVICES"] = args.gpus
python a.py --gpus 7
python a.py -G 7
argparse = argparse.ArgumentParser()
argparse.add_argument('gpus',type=str,help='using gpus',default='7')
args = argparse.parse_args()
os.environ["CUDA_VISIBLE_DEVICES"] = args.gpus
python a.py 7
网友评论