Skip to content

Commit 0cd50db

Browse files
piyu200594facebook-github-bot
authored andcommitted
Update README.md (#2173)
Summary: Updating the command to run python filr from python filename.py to python3 filename.py Pull Request resolved: #2173 Reviewed By: aaronenyeshi Differential Revision: D54092833 Pulled By: xuzhao9 fbshipit-source-id: f6ea9efd6e00f46e72d5134496ff74db62c23e87
1 parent 4c7b6ec commit 0cd50db

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

README.md

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -40,22 +40,22 @@ Install the benchmark suite, which will recursively install dependencies for all
4040
```
4141
git clone https://github.com/pytorch/benchmark
4242
cd benchmark
43-
python install.py
43+
python3 install.py
4444
```
4545

4646
### Install torchbench as a library
4747

4848
if you're interested in running torchbench as a library you can
4949

5050
```bash
51-
python install.py
51+
python3 install.py
5252
pip install git+https://www.github.com:pytorch/benchmark.git
5353
```
5454

5555
or
5656

5757
```bash
58-
python install.py
58+
python3 install.py
5959
pip install . # add -e for an editable installation
6060
```
6161

@@ -76,7 +76,7 @@ Then,
7676
```
7777
git clone https://github.com/pytorch/benchmark
7878
cd benchmark
79-
python install.py
79+
python3 install.py
8080
```
8181

8282
### Notes
@@ -111,11 +111,11 @@ There are multiple ways for running the model benchmarks.
111111
In each model repo, the assumption is that the user would already have all of the torch family of packages installed (torch, torchvision, torchaudio...) but it installs the rest of the dependencies for the model.
112112

113113
### Using `test.py`
114-
`python test.py` will execute the APIs for each model, as a sanity check. For benchmarking, use `test_bench.py`. It is based on unittest, and supports filtering via CLI.
114+
`python3 test.py` will execute the APIs for each model, as a sanity check. For benchmarking, use `test_bench.py`. It is based on unittest, and supports filtering via CLI.
115115

116116
For instance, to run the BERT model on CPU for the train execution mode:
117117
```
118-
python test.py -k "test_BERT_pytorch_train_cpu"
118+
python3 test.py -k "test_BERT_pytorch_train_cpu"
119119
```
120120

121121
The test name follows the following pattern:
@@ -147,7 +147,7 @@ The `userbenchmark` allows you to develop your customized benchmarks with TorchB
147147
Sometimes you may want to just run train or eval on a particular model, e.g. for debugging or profiling. Rather than relying on __main__ implementations inside each model, `run.py` provides a lightweight CLI for this purpose, building on top of the standard BenchmarkModel API.
148148

149149
```
150-
python run.py <model> [-d {cpu,cuda}] [-t {eval,train}] [--profile]
150+
python3 run.py <model> [-d {cpu,cuda}] [-t {eval,train}] [--profile]
151151
```
152152
Note: `<model>` can be a full, exact name, or a partial string match.
153153

0 commit comments

Comments
 (0)