Flexible version for torch

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Jul 8, 2024 · 59a71ba · 59a71ba
1 parent ce045ef
commit 59a71ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/required.txt b/requirements/required.txt
@@ -2,12 +2,12 @@ torchgeo==0.5.1
 rioxarray==0.15.0
 albumentations==1.3.1
 rasterio==1.3.9
-torch==2.1.0
+torch>=2.1.0
 torchvision==0.16.0
 torchmetrics==1.3.1
 geopandas==0.14.2
 lightly==1.4.25
 h5py==3.10.0
 geobench==1.0.0
 mlflow==2.12.1
-lightning==2.2.5
+lightning==2.2.5