Bump min version of torch

microsoft · Jun 29, 2023 · c5e70cd · c5e70cd
1 parent 2cff20c
commit c5e70cd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/environment.yml b/environment.yml
@@ -12,7 +12,7 @@ dependencies:
   - pycocotools>=2.0.4
   - pyproj>=3
   - python>=3.9
-  - pytorch>=1.12
+  - pytorch>=1.13
   - pyvista>=0.29
   - rarfile>=4
   - rasterio>=1.2

diff --git a/requirements/min-reqs.old b/requirements/min-reqs.old
@@ -16,7 +16,7 @@ rtree==1.0.0
 segmentation-models-pytorch==0.2.0
 shapely==1.7.1
 timm==0.4.12
-torch==1.12.0
+torch==1.13.0
 torchmetrics==0.10.0
 torchvision==0.13.0
 

diff --git a/setup.cfg b/setup.cfg
@@ -53,8 +53,8 @@ install_requires =
     shapely>=1.7.1,<3
     # timm 0.4.12 required by segmentation-models-pytorch
     timm>=0.4.12,<0.10
-    # torch 1.12+ required by torchvision
-    torch>=1.12,<3
+    # torch 1.13+ required for supporting percentages in random_split
+    torch>=1.13,<3
     # torchmetrics 0.10+ required for binary/multiclass/multilabel classification metrics
     torchmetrics>=0.10,<0.12
     # torchvision 0.13+ required for torchvision.models._api.WeightsEnum