Bump min version of torch

microsoft · Jul 12, 2023 · 2f8a82a · 2f8a82a
1 parent 4ce9ac8
commit 2f8a82a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/environment.yml b/environment.yml
@@ -12,12 +12,12 @@ dependencies:
   - pycocotools>=2.0.4
   - pyproj>=3
   - python>=3.9
-  - pytorch>=1.12
+  - pytorch>=1.13
   - pyvista>=0.29
   - rarfile>=4
   - rasterio>=1.2
   - shapely>=1.7.1
-  - torchvision>=0.13
+  - torchvision>=0.14
   - pip:
     - black[jupyter]>=21.8
     - flake8>=3.8

diff --git a/pyproject.toml b/pyproject.toml
@@ -65,12 +65,12 @@ dependencies = [
     "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,<2",
-    # torchvision 0.13+ required for torchvision.models._api.WeightsEnum
-    "torchvision>=0.13,<0.16",
+    # torchvision 0.14+ required by torch 1.13+
+    "torchvision>=0.14,<0.16",
 ]
 dynamic = ["version"]
 

diff --git a/requirements/min-reqs.old b/requirements/min-reqs.old
@@ -16,9 +16,9 @@ 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
+torchvision==0.14.0
 
 # datasets
 h5py==3.0.0