diff --git a/README.md b/README.md
index d6fa96155bc97d266da291d71170f6e7872eeee5..13d14122bec97f81ba74c57345faa0e8ac7b5a7f 100644
--- a/README.md
+++ b/README.md
@@ -27,13 +27,13 @@ Llama-recipes provides a pip distribution for easy install and usage in other pr
 
 ## Install with pip
 ```
-pip install llama-recipes
+pip install --extra-index-url https://download.pytorch.org/whl/test/cu118 llama-recipes
 ```
 ## Install from source
 To install from source e.g. for development use this command. We're using hatchling as our build backend which requires an up-to-date pip as well as setuptools package.
 ```
 pip install -U pip setuptools
-pip install -e .
+pip install -e --extra-index-url https://download.pytorch.org/whl/test/cu118 .
 ```
 ⚠️ **Note** ⚠️  Some features (especially fine-tuning with FSDP + PEFT) currently require PyTorch nightlies to be installed. Please make sure to install the nightlies if you're using these features following [this guide](https://pytorch.org/get-started/locally/).
 
diff --git a/requirements.txt b/requirements.txt
index ddda396d4beca2ca6e0f329926846d8e6c2601b1..5ce375234de0754dfcbfdb53f87b4ab9625a9e73 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
---index-url https://download.pytorch.org/whl/test/cu118 torch 
+torch>=2.0.1
 accelerate
 appdirs
 loralib