You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+3-3Lines changed: 3 additions & 3 deletions
Original file line number
Diff line number
Diff line change
@@ -159,20 +159,20 @@ Things we're excited about but need more time to cook in the oven
159
159
160
160
`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.
161
161
162
-
Stable release from Pypi which will default to CUDA 12.1
162
+
Stable release from Pypi which will default to CUDA 12.4
163
163
164
164
```Shell
165
165
pip install torchao
166
166
```
167
167
168
168
Stable Release from the PyTorch index
169
169
```Shell
170
-
pip install torchao --extra-index-url https://download.pytorch.org/whl/cu121# full options are cpu/cu118/cu121/cu124
170
+
pip install torchao --extra-index-url https://download.pytorch.org/whl/cu124# full options are cpu/cu118/cu124/cu126
171
171
```
172
172
173
173
Nightly Release
174
174
```Shell
175
-
pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu121# full options are cpu/cu118/cu121/cu124
175
+
pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu126# full options are cpu/cu118/cu126/cu128
176
176
```
177
177
178
178
For *most* developers you probably want to skip building custom C++/CUDA extensions for faster iteration
0 commit comments