From 7046d1b39b209302c9eabf9e1c3e1e73bfdc1e6c Mon Sep 17 00:00:00 2001 From: healthonrails Date: Mon, 14 Oct 2024 14:41:45 -0400 Subject: [PATCH] Update torch and torchvison versions --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index f2862cc..f6ed58e 100644 --- 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==2.4.0+cpu +torch>=2.4.0 --find-links https://download.pytorch.org/whl/torch_stable.html -torchvision==0.19.0+cpu +torchvision>=0.19.0 decord>=0.4.0 opencv-python>=4.1.2.30 opencv-contrib-python>=4.1.2.30