diff --git a/requirements/runtime.txt b/requirements/runtime.txt
index 6eb7cafcc..2ec3cdce7 100644
--- a/requirements/runtime.txt
+++ b/requirements/runtime.txt
@@ -1,4 +1,3 @@
---index-url https://download.pytorch.org/whl/cu121
 albumentations==1.3.0
 ansi2html==1.8.0
 # apispec==0.38.0
@@ -80,9 +79,9 @@ SQLAlchemy-Utils==0.38.3
 tensorboard_logger==0.1.0
 Theano==1.0.5
 timm==0.6.12
-torch==2.1.0
-torchvision==0.16.0
-torchaudio==2.1.0
+torch==2.1.0 --index-url https://download.pytorch.org/whl/cu121
+torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu121
+torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
 tornado==6.2
 tqdm==4.64.0
 ubelt==1.2.1