From 2f8a82a92da1fe5044e49c6319ee812dfe5e00a4 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 | 4 ++-- pyproject.toml | 8 ++++---- requirements/min-reqs.old | 4 ++-- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/environment.yml b/environment.yml index 1ffcbb950f0..e72e9027e9b 100644 --- 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 index d1d09b442da..d61481762da 100644 --- 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 index 18f0890b2de..986e1d8515d 100644 --- 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