Build for PyTorch 2.1 (fairinternal/xformers#822)

* Build for PyTorch 2.1 * pt2.1 does not exist with cu117 --------- Co-authored-by: danthe3rd <danthe3rd> __original_commit__ = fairinternal/xformers@cbb7c34
facebookresearch · Oct 6, 2023 · d20354a · d20354a
1 parent 3dafaf4
commit d20354a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 12 deletions.
diff --git a/.github/workflows/conda.yml b/.github/workflows/conda.yml
@@ -34,7 +34,7 @@ jobs:
           - "3.10"
         config:
           - torch_version: "2.1.0"
-            torch_channel: "pytorch-nightly"
+            torch_channel: "pytorch"
             cuda_version: "11.8.0"
             cuda_dep_runtime: ">=11.7,<11.9"
             cuda_short_version: "118"
@@ -51,12 +51,6 @@ jobs:
             cuda_dep_runtime: ">=11.6,<11.8"
             cuda_short_version: "116"
 
-          - torch_version: "1.12.1"
-            torch_channel: "pytorch"
-            cuda_version: "11.6.2"
-            cuda_dep_runtime: ">=11.3,<11.7"
-            cuda_short_version: "116"
-
     name: py${{ matrix.python }}-torch${{ matrix.config.torch_version }}-cu${{ matrix.config.cuda_version }}
     runs-on: 4-core-ubuntu  # 16GB RAM, 4 vCPUs
 

diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -27,21 +27,21 @@ jobs:
           - "3.10"
           - "3.11"
         torch_version:
-          - "2.0.1"
+          - "2.1.0"
         include:
           - sdist: false
             publish: false
             cuda_short_version: "CUDA_VERSION_NOT_SET"
           # Publish only for latest pytorch
-          - torch_version: "2.0.1"
+          - torch_version: "2.1.0"
             publish: true
           # Publish source distribution only from this runner
           - os: 4-core-ubuntu
             python: "3.10"
-            torch_version: "2.0.1"
+            torch_version: "2.1.0"
             sdist: true
           # Set corresponding CUDA version
-          - torch_version: "2.0.1"
+          - torch_version: "2.1.0"
             cuda_short_version: "118"
 
     uses: ./.github/workflows/wheels_reusable.yml

diff --git a/.github/workflows/win-build.yml b/.github/workflows/win-build.yml
@@ -48,7 +48,7 @@ jobs:
 
       - name: Install build dependencies
         run: |
-          $PY -m pip install wheel setuptools ninja torch==2.0.1 -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu117
+          $PY -m pip install wheel setuptools ninja torch==2.1.0 -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118
           git config --global --add safe.directory "*"
           $PY -c "import torch; print('torch', torch.__version__)"
           $PY -c "import torch; print('torch.cuda', torch.version.cuda)"