diff --git a/requirements/torch.txt b/requirements/torch.txt
new file mode 100644
index 000000000..92bee1954
--- /dev/null
+++ b/requirements/torch.txt
@@ -0,0 +1,4 @@
+--index-url https://download.pytorch.org/whl/cu121
+torch==2.1.0 
+torchvision==0.16.0
+torchaudio==2.1.0