Skip to content

Commit a4fdd4d

Browse files
authored
Add dep installation CLI command (#15103)
1 parent 2357c63 commit a4fdd4d

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

paddleocr/_cli.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@
1313
# limitations under the License.
1414

1515
import argparse
16+
import subprocess
17+
import sys
1618
import warnings
1719

1820
from ._models import (
@@ -79,12 +81,27 @@ def _register_models(subparsers):
7981
subparser.set_defaults(executor=subcommand_executor.execute_with_args)
8082

8183

84+
def _register_install_hpi_deps_command(subparsers):
85+
def _install_hpi_deps(args):
86+
hpip = f"hpi-{args.variant}"
87+
try:
88+
subprocess.check_call(["paddlex", "--install", hpip])
89+
subprocess.check_call(["paddlex", "--install", "paddle2onnx"])
90+
except subprocess.CalledProcessError:
91+
sys.exit("Failed to install dependencies")
92+
93+
subparser = subparsers.add_parser("install_hpi_deps")
94+
subparser.add_argument("variant", type=str, choices=["cpu", "gpu", "npu"])
95+
subparser.set_defaults(executor=_install_hpi_deps)
96+
97+
8298
def _parse_args():
8399
parser = argparse.ArgumentParser(prog="paddleocr")
84100
parser.add_argument("--version", action="version", version=f"%(prog)s {version}")
85101
subparsers = parser.add_subparsers(dest="subcommand")
86102
_register_pipelines(subparsers)
87103
_register_models(subparsers)
104+
_register_install_hpi_deps_command(subparsers)
88105
return parser.parse_args()
89106

90107

0 commit comments

Comments
 (0)