We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c8d9fa5 commit 335171fCopy full SHA for 335171f
README.md
@@ -18,9 +18,12 @@ pip install torchao
18
```
19
20
Nightly Release
21
-```Shell
22
-pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/ # CUDA builds
+```Shell
23
pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cpu # CPU only builds
+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
+
27
28
29
From source
0 commit comments