add upgrade to pip command (#1687)

Co-authored-by: Felipe Mello <felipemello@fb.com>
pytorch · Sep 25, 2024 · 4e69db8 · 4e69db8
1 parent aa5c90e
commit 4e69db8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -119,8 +119,8 @@ pip install torchtune
 
 ```bash
 # Install PyTorch, torchvision, torchao nightlies
-pip install --pre torch torchvision torchao --index-url https://download.pytorch.org/whl/nightly/cu121
-pip install --pre torchtune --extra-index-url https://download.pytorch.org/whl/nightly/cpu
+pip install --pre --upgrade torch torchvision torchao --index-url https://download.pytorch.org/whl/nightly/cu121
+pip install --pre --upgrade torchtune --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 ```
 
 You can also check out our [install documentation](https://pytorch.org/torchtune/main/install.html) for more information, including installing torchtune from source.