Skip to content

Commit

Permalink
Update installation instructions (#447)
Browse files Browse the repository at this point in the history
* Update installation instructions

* Update README.md
  • Loading branch information
msaroufim authored Jun 26, 2024
1 parent f56e272 commit 2e3ec26
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
```
Expand Down

0 comments on commit 2e3ec26

Please sign in to comment.