From 17a46eac48175a67c206222ead0b157b39f687d2 Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Fri, 28 Jun 2024 10:20:41 -0700 Subject: [PATCH 1/4] Update installation --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 7b03d402ed..77b0605376 100644 --- a/README.md +++ b/README.md @@ -83,7 +83,7 @@ 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 +Stable release from Pypi which will default to CUDA 12.1 ```Shell pip install torchao @@ -91,7 +91,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 From 48a8ef43dc8f5b30a3ce02e6d96390734b75376c Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Fri, 28 Jun 2024 10:23:06 -0700 Subject: [PATCH 2/4] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 77b0605376..c55e17e57a 100644 --- a/README.md +++ b/README.md @@ -83,6 +83,8 @@ 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. +Install torch from https://pytorch.org/ + Stable release from Pypi which will default to CUDA 12.1 ```Shell From 4f4bb03ff2b4405c15bbe2ec5c23a3050b221d7c Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Fri, 28 Jun 2024 11:22:11 -0700 Subject: [PATCH 3/4] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index c55e17e57a..0ec608af36 100644 --- a/README.md +++ b/README.md @@ -83,7 +83,7 @@ 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. -Install torch from https://pytorch.org/ +Install the torch version you want from https://pytorch.org/ Stable release from Pypi which will default to CUDA 12.1 From 830f374273fbaea36b819d9368a9a9b6b5ddadd1 Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Fri, 28 Jun 2024 11:47:44 -0700 Subject: [PATCH 4/4] Update README.md --- README.md | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 0ec608af36..d4b4ca3bbc 100644 --- a/README.md +++ b/README.md @@ -81,9 +81,24 @@ 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. -Install the torch version you want from https://pytorch.org/ +#### 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