diff --git a/requirements.txt b/requirements.txt
index ce07c518..06a20ee7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch # We have to place it here otherwise it installs the CPU version
+torch==2.0.1 # We have to place it here otherwise it installs the CPU version
 Pillow>=10.0.1
 pydantic==1.10.12
 horde_model_reference~=0.2.2