Skip to content

Commit 335171f

Browse files
authored
Add architecture to index postfix for nightly builds (#336)
1 parent c8d9fa5 commit 335171f

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

README.md

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,12 @@ pip install torchao
1818
```
1919

2020
Nightly Release
21-
```Shell
22-
pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/ # CUDA builds
21+
```Shell
2322
pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cpu # CPU only builds
23+
pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cu118 # CUDA 11.8
24+
pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cu121 # CUDA 12.1
25+
pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cu124 # CUDA 12.4
26+
2427
```
2528

2629
From source

0 commit comments

Comments
 (0)