Update installation instructions (pytorch#447)

* Update installation instructions * Update README.md
dbyoung18 · Jun 26, 2024 · 422e2d6 · 422e2d6
1 parent d63cbfe
commit 422e2d6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -83,7 +83,13 @@ A key design principle for us is composability as in any new dtype or layout we
 ### Installation
 `torchao` makes liberal use of several new features in Pytorch, it's recommended to use it with the current nightly or latest stable version of PyTorch.
 
-Stable Release
+Stable release from Pypi which will default to PyTorch 2.3.1 and CUDA 12.1
+
+```Shell
+pip install torchao
+```
+
+Stable Release from the PyTorch index
 ```Shell
 pip install torchao --extra-index-url https://download.pytorch.org/whl/test/cu121 # full options are cpu/cu118/cu121/cu124
 ```