Update runtime.txt

WildMeOrg · Jun 15, 2024 · acc2726 · acc2726
1 parent f398ba3
commit acc2726
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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