Merge pull request #233 from WildMeOrg/cuda-version-fix

The additional cuda packages removed
WildMeOrg · Mar 24, 2023 · d14a1ed · d14a1ed
2 parents 6e69686 + 6405cc4
commit d14a1ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/runtime.txt b/requirements/runtime.txt
@@ -74,8 +74,8 @@ sqlalchemy==1.4.14
 SQLAlchemy-Utils==0.38.3
 tensorboard_logger==0.1.0
 Theano==1.0.5
-torch==1.12.1+cu113
-torchvision==0.13.1+cu113
+torch==1.12.1
+torchvision==0.13.1
 torchaudio==0.12.1
 tornado==6.2
 tqdm==4.64.0