Bumped PyTorch version to 1.13.1

Unity-Technologies · Oct 5, 2023 · 0ef356a · 0ef356a
1 parent 42ab41f
commit 0ef356a
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 8 deletions.
diff --git a/.github/workflows/nightly.yml b/.github/workflows/nightly.yml
@@ -40,7 +40,7 @@ jobs:
         python-version: [3.10.x]
         include:
           - python-version: 3.10.x
-            pip_constraints: test_constraints_max_version.txt
+            pip_constraints: test_constraints_version.txt
     steps:
     - uses: actions/checkout@v2
     - name: Set up Python

diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -40,7 +40,7 @@ jobs:
         python-version: [3.10.12]
         include:
           - python-version: 3.10.12
-            pip_constraints: test_constraints_max_version.txt
+            pip_constraints: test_constraints_version.txt
     steps:
     - uses: actions/checkout@v2
     - name: Set up Python

diff --git a/ml-agents/setup.py b/ml-agents/setup.py
@@ -62,7 +62,7 @@ def run(self):
         "Pillow>=4.2.1",
         "protobuf>=3.6,<3.20",
         "pyyaml>=3.1.0",
-        "torch>=1.8.0,<=1.11.0",
+        "torch>=1.13.1",
         "tensorboard>=2.14",
         # adding six explicit dependency since tensorboard needs it but doesn't declare it as a dep
         "six>=1.16",
@@ -72,6 +72,7 @@ def run(self):
         "attrs>=19.3.0",
         "huggingface_hub>=0.14",
         'pypiwin32==223;platform_system=="Windows"',
+        "onnx==1.12.0",
     ],
     python_requires=">=3.10.1,<=3.10.12",
     entry_points={

diff --git a/test_constraints_mid_version.txt b/test_constraints_mid_version.txt
diff --git a/test_constraints_min_version.txt b/test_constraints_min_version.txt
diff --git a/test_constraints_max_version.txt → test_constraints_version.txt b/test_constraints_max_version.txt → test_constraints_version.txt
@@ -1,3 +1,3 @@
 # pip constraints to use the *highest* versions allowed in ml-agents/setup.py
 # For projects with upper bounds, we should periodically update this list to the latest
-torch==1.11.0
+torch==1.13.1