Update torch requirement from <1.14,>=1.9 to >=1.9,<2.1 (#539)

Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Dirk Groeneveld <[email protected]>
allenai · May 8, 2023 · 70f2681 · 70f2681
1 parent 1ee2c56
commit 70f2681
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -205,7 +205,7 @@ jobs:
         if: steps.virtualenv-cache.outputs.cache-hit != 'true' && (contains(matrix.task.extras, 'torch') || contains(matrix.task.extras, 'all') || matrix.task.requires_torch)
         run: |
           . .venv/bin/activate
-          pip install torch==1.13.0 --extra-index-url https://download.pytorch.org/whl/cpu
+          pip install torch==2.0.0 --extra-index-url https://download.pytorch.org/whl/cpu
 
       - name: Pre-install flax
         if: steps.virtualenv-cache.outputs.cache-hit != 'true' && (contains(matrix.task.extras, 'flax') || contains(matrix.task.extras, 'all'))

diff --git a/Dockerfile b/Dockerfile
@@ -1,6 +1,6 @@
 # This Dockerfile can be used to build a Docker image suitable for tango projects.
 
-ARG BASE_IMAGE=ghcr.io/allenai/pytorch:1.13.0-cuda11.6-python3.9
+ARG BASE_IMAGE=ghcr.io/allenai/pytorch:2.0.0-cuda11.7-python3.10
 FROM ${BASE_IMAGE}
 
 WORKDIR /stage

diff --git a/Dockerfile.test b/Dockerfile.test
@@ -4,7 +4,7 @@
 # variable "COMMIT_SHA". That way we don't need to rebuild and push the image each time we run
 # tests, and we can be sure the dependencies are always up-to-date.
 
-FROM ghcr.io/allenai/pytorch:1.13.0-cuda11.6-python3.9
+FROM ghcr.io/allenai/pytorch:2.0.0-cuda11.7-python3.10
 
 COPY scripts/entrypoint.sh /entrypoint.sh
 RUN chmod +x /entrypoint.sh

diff --git a/pyproject.toml b/pyproject.toml
@@ -68,11 +68,11 @@ examples = [
   "torchmetrics>=0.7.0"
 ]
 torch = [
-  "torch>=1.9,<1.14",
+  "torch>=1.9,<2.1",
   "numpy",
 ]
 transformers = [
-  "torch>=1.9,<1.14",
+  "torch>=1.9,<2.1",
   "numpy",
   "datasets>=1.12,<3.0",
   "transformers>=4.12.3",
@@ -83,7 +83,7 @@ datasets = [
   "datasets>=1.12,<3.0"
 ]
 fairscale = [
-  "torch>=1.9,<1.14",
+  "torch>=1.9,<2.1",
   "numpy",
   "fairscale>=0.4.6,<0.5"
 ]