From 41083037a4844f8070061dbbcc5c903f98811fd8 Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Fri, 28 Jun 2024 11:58:02 -0700 Subject: [PATCH] Update installation Docs (#459) * Update installation * Update README.md * Update README.md * Update README.md --- README.md | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 7b03d402ed..d4b4ca3bbc 100644 --- a/README.md +++ b/README.md @@ -81,9 +81,26 @@ 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 from Pypi which will default to PyTorch 2.3.1 and CUDA 12.1 +#### Install torch + +Install torch stable + +``` +pip install torch +``` + +Or torch nightlies + +``` +pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 +``` + +#### Install torchao + +Stable release from Pypi which will default to CUDA 12.1 ```Shell pip install torchao @@ -91,7 +108,7 @@ 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 +pip install torchao --extra-index-url https://download.pytorch.org/whl/cu121 # full options are cpu/cu118/cu121/cu124 ``` Nightly Release