From c5e70cd98b7ca19c6ecb47e43bd7b7e34ad1225a Mon Sep 17 00:00:00 2001 From: Ashwin Nair Date: Thu, 29 Jun 2023 14:36:24 +0400 Subject: [PATCH] Bump min version of torch --- environment.yml | 2 +- requirements/min-reqs.old | 2 +- setup.cfg | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/environment.yml b/environment.yml index de47c5cc647..5b1faea5163 100644 --- 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 index 84375103261..7e657ee0805 100644 --- 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 index 4e8553e8358..6e5235d9aff 100644 --- 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