[Build] update minor version & update torch to >= 2.0 (#1747)

mindee · Oct 10, 2024 · 7f21e76 · 7f21e76
1 parent 0ca8249
commit 7f21e76
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 8 deletions.
diff --git a/.conda/meta.yaml b/.conda/meta.yaml
@@ -1,7 +1,7 @@
 {% set pyproject = load_file_data('../pyproject.toml', from_recipe_dir=True) %}
 {% set project = pyproject.get('project') %}
 {% set urls = pyproject.get('project', {}).get('urls') %}
-{% set version = environ.get('BUILD_VERSION', '0.9.1a0') %}
+{% set version = environ.get('BUILD_VERSION', '0.10.0a0') %}
 
 package:
   name: {{ project.get('name') }}

diff --git a/README.md b/README.md
@@ -161,7 +161,7 @@ pip install "python-doctr[torch,viz,html,contib]"
 For MacBooks with M1 chip, you will need some additional packages or specific versions:
 
 - TensorFlow 2: [metal plugin](https://developer.apple.com/metal/tensorflow-plugin/)
-- PyTorch: [version >= 1.12.0](https://pytorch.org/get-started/locally/#start-locally)
+- PyTorch: [version >= 2.0.0](https://pytorch.org/get-started/locally/#start-locally)
 
 ### Developer mode
 

diff --git a/docs/source/getting_started/installing.rst b/docs/source/getting_started/installing.rst
@@ -17,7 +17,7 @@ Whichever OS you are running, you will need to install at least TensorFlow or Py
 For MacBooks with M1 chip, you will need some additional packages or specific versions:
 
 * `TensorFlow 2 Metal Plugin <https://developer.apple.com/metal/tensorflow-plugin/>`_
-* `PyTorch >= 1.12.0 <https://pytorch.org/get-started/locally/#start-locally>`_
+* `PyTorch >= 2.0.0 <https://pytorch.org/get-started/locally/#start-locally>`_
 
 Via Python Package
 ==================

diff --git a/pyproject.toml b/pyproject.toml
@@ -58,8 +58,8 @@ tf = [
     "tf2onnx>=1.16.0,<2.0.0",  # cf. https://github.com/onnx/tensorflow-onnx/releases/tag/v1.16.0
 ]
 torch = [
-    "torch>=1.12.0,<3.0.0",
-    "torchvision>=0.13.0",
+    "torch>=2.0.0,<3.0.0",
+    "torchvision>=0.15.0",
     "onnx>=1.12.0,<3.0.0",
 ]
 html = [
@@ -101,8 +101,8 @@ dev = [
     "tf-keras>=2.15.0,<3.0.0",  # Keep keras 2 compatibility
     "tf2onnx>=1.16.0,<2.0.0",  # cf. https://github.com/onnx/tensorflow-onnx/releases/tag/v1.16.0
     # PyTorch
-    "torch>=1.12.0,<3.0.0",
-    "torchvision>=0.13.0",
+    "torch>=2.0.0,<3.0.0",
+    "torchvision>=0.15.0",
     "onnx>=1.12.0,<3.0.0",
     # Extras
     "weasyprint>=55.0",

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 from setuptools import setup
 
 PKG_NAME = "python-doctr"
-VERSION = os.getenv("BUILD_VERSION", "0.9.1a0")
+VERSION = os.getenv("BUILD_VERSION", "0.10.0a0")
 
 
 if __name__ == "__main__":