Merge pull request #27 from Starrah/patch-pytorch-version

优化对torch版本的依赖声明为>=1.12.1且<2.0
iMoonLab · Dec 21, 2022 · 2b25d32 · 2b25d32
2 parents 53055cc + a4c7e46
commit 2b25d32
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docs/source/start/install.rst b/docs/source/start/install.rst
@@ -7,7 +7,7 @@ Installation
     - Proof: Xinwei Zhang
 
 - Python >= 3.8
-- Pytorch >=1.11
+- Pytorch >= 1.12
 
 Current, the stable version of **DHG** is 0.9.2. You can install it with ``pip`` as follows:
 

diff --git a/docs/source/zh/start/install.rst b/docs/source/zh/start/install.rst
@@ -8,7 +8,7 @@
     - 校对： `丰一帆 <https://fengyifan.site/>`_ 、张欣炜
 
 - Python >= 3.8
-- Pytorch >=1.11
+- Pytorch >= 1.12
 
 
 目前， **DHG** 的最新稳定版本 **0.9.2** 已经发布，可以使用 ``pip`` 指令直接安装：

diff --git a/pyproject.toml b/pyproject.toml
@@ -32,7 +32,7 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = "^3.8"
-torch = "1.12.1"
+torch = ">= 1.12.1, < 2.0"
 scipy = "^1.8"
 optuna = "*"
 numpy = "*"