diff --git a/install.py b/install.py index ea825e6f7..943de9dc1 100644 --- a/install.py +++ b/install.py @@ -55,6 +55,8 @@ def pip_install_requirements(requirements_txt="requirements.txt"): end="", flush=True, ) + if args.userbenchmark: + TORCH_DEPS = ["torch"] try: versions = get_pkg_versions(TORCH_DEPS) except ModuleNotFoundError as e: