Sync torch and torchvision versions with pytorch-lightning

ref https://github.com/Lightning-AI/lightning/blob/2.0.2/requirements/pytorch/base.txt#L5 ref #3190 (comment)
autogluon · May 5, 2023 · 4c94b71 · 4c94b71
1 parent cdbd78e
commit 4c94b71
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/multimodal/setup.py b/multimodal/setup.py
@@ -33,8 +33,8 @@
     "evaluate>=0.2.2,<0.4.0",
     "accelerate>=0.9,<0.17",
     "timm>=0.6.12,<0.7.0",
-    "torch>=1.9,<2.1",
-    "torchvision>=0.10.0,<0.16",
+    "torch>=1.11,<2.1",
+    "torchvision>=0.12.0,<0.16",
     "fairscale>=0.4.5,<0.4.14",
     "scikit-image>=0.19.1,<0.21.0",
     "pytorch-lightning>=2.0.0,<2.1",

diff --git a/tabular/setup.py b/tabular/setup.py
@@ -43,7 +43,7 @@
         'xgboost>=1.6,<1.8',
     ],
     'fastai': [
-        'torch>=1.9,<2.1',
+        'torch>=1.11,<2.1',
         'fastai>=2.3.1,<2.8',
     ],
     'ray': [

diff --git a/timeseries/setup.py b/timeseries/setup.py
@@ -29,7 +29,7 @@
     "pandas",  # version range defined in `core/_setup_utils.py`
     "statsmodels>=0.13.0,<0.14",
     "gluonts>=0.12.4,<0.13",
-    "torch>=1.9,<2.1",
+    "torch>=1.11,<2.1",
     "pytorch-lightning>=2.0.0,<2.1",
     "networkx",  # version range defined in `core/_setup_utils.py`
     "statsforecast>=1.4.0,<1.5",