diff --git a/README.md b/README.md
index d622c9072..979966a82 100644
--- a/README.md
+++ b/README.md
@@ -95,16 +95,16 @@ Note: pytorch does not support python 3.12 yet so make sure your python version
 ### AMD GPUs (Linux only)
 AMD users can install rocm and pytorch with pip if you don't have it already installed, this is the command to install the stable version:
 
-```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.6```
+```pip install -r requirements-rocm5.6.txt```
 
 This is the command to install the nightly with ROCm 5.7 that might have some performance improvements:
-```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.7```
+```pip install -r requirements-rocm5.7-nightly.txt```
 
 ### NVIDIA
 
 Nvidia users should install pytorch using this command:
 
-```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121```
+```pip install -r requirements-cu121.txt```
 
 #### Troubleshooting
 
diff --git a/requirements-cpu.txt b/requirements-cpu.txt
new file mode 100644
index 000000000..30ed93f6e
--- /dev/null
+++ b/requirements-cpu.txt
@@ -0,0 +1,5 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+
+torch
+torchvision
+torchaudio
\ No newline at end of file
diff --git a/requirements-cu118.txt b/requirements-cu118.txt
new file mode 100644
index 000000000..7f824cb2f
--- /dev/null
+++ b/requirements-cu118.txt
@@ -0,0 +1,5 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
+
+torch
+torchvision
+torchaudio
\ No newline at end of file
diff --git a/requirements-cu121.txt b/requirements-cu121.txt
new file mode 100644
index 000000000..4040a64dd
--- /dev/null
+++ b/requirements-cu121.txt
@@ -0,0 +1,5 @@
+--extra-index-url https://download.pytorch.org/whl/cu121
+
+torch
+torchvision
+torchaudio
\ No newline at end of file
diff --git a/requirements-rocm5.6.txt b/requirements-rocm5.6.txt
new file mode 100644
index 000000000..9187bf064
--- /dev/null
+++ b/requirements-rocm5.6.txt
@@ -0,0 +1,5 @@
+--extra-index-url https://download.pytorch.org/whl/rocm5.6
+
+torch
+torchvision
+torchaudio
\ No newline at end of file
diff --git a/requirements-rocm5.7-nightly.txt b/requirements-rocm5.7-nightly.txt
new file mode 100644
index 000000000..e8a5d3e06
--- /dev/null
+++ b/requirements-rocm5.7-nightly.txt
@@ -0,0 +1,6 @@
+--pre
+--index-url https://download.pytorch.org/whl/nightly/rocm5.7
+
+torch
+torchvision
+torchaudio
\ No newline at end of file