diff --git a/.conda/meta.yaml b/.conda/meta.yaml
index 7feb3a1bf..f377481f1 100644
--- 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
index 2fc92971f..d57228fda 100644
--- 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
index 46d4177b3..e764e734a 100644
--- 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 `_
-* `PyTorch >= 1.12.0 `_
+* `PyTorch >= 2.0.0 `_
Via Python Package
==================
diff --git a/pyproject.toml b/pyproject.toml
index 9745f8a7c..613eb512e 100644
--- 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
index f45f3f157..13fd4515e 100644
--- 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__":