doc

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