Skip to content

Commit 891e0c0

Browse files
committed
Change install everything to use pip
.. instead of editing PYTHONPATH. The latter is not persistent
1 parent 04f1401 commit 891e0c0

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

README.md

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,9 @@ git clone https://github.com/pytorch-tpu/jetstream-pytorch.git
1414
### 2. Run installation script:
1515

1616
```bash
17-
source install_everything.sh
17+
sh install_everything.sh
1818
```
1919

20-
NOTE: the above script will export `PYTHONPATH`, so sourcing will make it
21-
to take effect in the current shell
22-
2320

2421
# Get weights
2522

install_everything.sh

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,10 @@ mkdir -p deps
99
pushd deps
1010
git clone https://github.com/google/JetStream.git
1111
git clone https://github.com/pytorch/xla.git
12-
pushd xla
12+
pushd xla/experimental/torch_xla2
1313
git checkout $TORCHXLA_TAG
14-
popd
14+
pip install .
1515
popd # now at the folder of jetstream-pytorch
16-
17-
export PYTHONPATH=$PYTHONPATH:$(pwd)/deps/xla/experimental/torch_xla2:$(pwd)/deps/JetStream:$(pwd)
18-
16+
pushd JetStream
17+
pip install .
18+
popd

0 commit comments

Comments
 (0)