Skip to content

Commit 055bcce

Browse files
authored
Update BKC for torchbench (#130)
1 parent 87d0b49 commit 055bcce

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

docs/test_docs/end_to_end_tests.md

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,6 @@ error: torch 2.1.0a0+gitdd9913f is installed but torch==2.1.0 is required by {'t
8383

8484
### Install Torch Audio
8585
```Bash
86-
pip install torchaudio
8786
git clone --recursive https://github.com/pytorch/audio.git
8887
cd audio
8988
python setup.py install
@@ -99,11 +98,15 @@ python -c "import torchvision,torchtext,torchaudio;print(torchvision.__version__
9998
Then install TorchBenchmark as a library:
10099
```
101100
conda install git-lfs pyyaml pandas scipy psutil
101+
pip install pyre_extensions
102+
pip install torchrec
103+
102104
git clone --recursive https://github.com/pytorch/benchmark.git
103105
104106
cd benchmark
105107
python install.py
106-
pip install .
108+
# Note that -e is necessary
109+
pip install -e .
107110
```
108111

109112
# Run the Model

0 commit comments

Comments
 (0)