Add architecture to index postfix for nightly builds (#336)

pytorch · Jun 7, 2024 · 335171f · 335171f
1 parent c8d9fa5
commit 335171f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -18,9 +18,12 @@ pip install torchao
 ```
 
 Nightly Release
-```Shell 
-pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/ # CUDA builds
+```Shell
 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
+pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cu121 # CUDA 12.1
+pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cu124 # CUDA 12.4
+
 ```
 
 From source