diff --git a/setup.py b/setup.py
index a57e5ae3..22ad27d9 100644
--- a/setup.py
+++ b/setup.py
@@ -59,7 +59,7 @@ def finalize_options(self):
         "h5py==3.11.0",
         "omegaconf==2.3.0",
         "torch>=2.2.0",
-        "torchvision==0.18.0",
+        "torchvision",
         "scikit-image>=0.19.0",
         "scikit-learn>=1.0.1",
         "tensorboard>=2.7.0",