diff --git a/pyproject.toml b/pyproject.toml
index 7e22fcf..01565aa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,12 +8,12 @@ packages = [{include = "auton_survival"}]
 
 [tool.poetry.dependencies]
 python = "^3.8"
-torch = "^1.13"
+torch = ">=1.13"
 numpy = "^1.24"
 pandas = "^1.5"
 tqdm = "^4.66"
 scikit-learn = "^1.2"
-torchvision = "^0.14"
+torchvision = ">=0.14"
 scikit-survival = "^0.21"
 lifelines = "^0.26"