Update nightly build instructions (pytorch#334)

dbyoung18 · Jun 7, 2024 · 199ac0a · 199ac0a
1 parent ebeb77a
commit 199ac0a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -18,8 +18,9 @@ pip install torchao
 ```
 
 Nightly Release
-```Shell
-pip install torchao-nightly
+```Shell 
+pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/ # CUDA builds
+pip install --pre torchao-nightly --index-url https://download.pytorch.org/whl/nightly/cpu # CPU only builds
 ```
 
 From source