diff --git a/requirements.txt b/requirements.txt
index e1c6607..3ff9b32 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,10 @@
 huggingface_hub
 ultralytics
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.3.0+cu121
+torchvision==0.18.0+cu121
+torchaudio==2.3.0+cu121
 albumentations
-
 opencv-python
 matplotlib
 PyYAML