Skip to content

Commit

Permalink
Remove torchvision and torchaudio dependency when testing userbenchma…
Browse files Browse the repository at this point in the history
…rk. (#2309)

Summary:
Most userbenchmarks handle dependencies themselves, and torchvision/torchaudio are not required.

Pull Request resolved: #2309

Reviewed By: aaronenyeshi

Differential Revision: D58686618

Pulled By: xuzhao9

fbshipit-source-id: b6ad66333586e237ef115c2c39c0069e4cb58450
  • Loading branch information
xuzhao9 authored and facebook-github-bot committed Jun 17, 2024
1 parent 51eca7b commit f5d3de1
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions install.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit f5d3de1

Please sign in to comment.