benchmark
9ff23aba - Add install.py for test_bench (#2546)

Commit
1 year ago
Add install.py for test_bench (#2546) Summary: Add install.py for test_bench introduced in https://github.com/pytorch/benchmark/pull/2052 ### Usage ```bash $ python install.py --userbenchmark test_bench --models BERT_pytorch hf_GPT2 --skip hf_GPT2 checking packages numpy, torch are installed, generating constaints...OK Installing userbenchmark test_bench with extra args: ['--models'] Installing BERT_pytorch... Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple Obtaining file:///Users/shenke/workspace/benchmark/torchbenchmark/models/BERT_pytorch Preparing metadata (setup.py) ... done Requirement already satisfied: tqdm in /opt/anaconda3/envs/nightlypth/lib/python3.11/site-packages (from bert_pytorch==0.0.1a4) (4.66.4) Requirement already satisfied: numpy in /opt/anaconda3/envs/nightlypth/lib/python3.11/site-packages (from bert_pytorch==0.0.1a4) (1.24.4) Installing collected packages: bert_pytorch Attempting uninstall: bert_pytorch Found existing installation: bert_pytorch 0.0.1a4 Uninstalling bert_pytorch-0.0.1a4: Successfully uninstalled bert_pytorch-0.0.1a4 Running setup.py develop for bert_pytorch Successfully installed bert_pytorch-0.0.1a4 ``` Pull Request resolved: https://github.com/pytorch/benchmark/pull/2546 Reviewed By: xuzhao9 Differential Revision: D66457458 Pulled By: FindHao fbshipit-source-id: 73b5f88dc50bd27eceb91c456279d7a687656c7c
Author
Parents
Loading