diff --git a/deepmd/entrypoints/main.py b/deepmd/entrypoints/main.py index 52e4e1c61e..5789cfe47b 100644 --- a/deepmd/entrypoints/main.py +++ b/deepmd/entrypoints/main.py @@ -5,6 +5,7 @@ from pathlib import Path from typing import Dict, List, Optional +from deepmd import __version__ from deepmd.entrypoints import ( compress, config, @@ -395,6 +396,8 @@ def parse_args(args: Optional[List[str]] = None): type=str, help='the output model', ) + # --version + parser.add_argument('--version', action='version', version='DeePMD-kit v%s' % __version__) parsed_args = parser.parse_args(args=args) if parsed_args.command is None: