Update PyTorch and torchvision versions in requirements.txt

healthonrails · Oct 14, 2024 · 45a32c5 · 45a32c5
1 parent 35dda01
commit 45a32c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,9 @@
 pyyaml>=5.3
 easydict>=1.9
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch==1.4.0+cpu
+torch==2.4.0+cpu
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torchvision==0.5.0+cpu
+torchvision==0.19.0+cpu
 decord>=0.4.0
 opencv-python>=4.1.2.30
 opencv-contrib-python>=4.1.2.30