Change torch version in pip build depending on the ci's version matrix

openmm · Jun 29, 2023 · f0f3bf0 · f0f3bf0
1 parent cc307f2
commit f0f3bf0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -118,13 +118,15 @@ jobs:
       if: ${{ !matrix.enable_cuda }}
       run: |
         conda activate nnpops
+        sed -i -e '/"torch/c "torch==${{ matrix.pytorch }}",' pyproject.toml
         pip -vvv wheel --wheel-dir=$(pwd) . --extra-index-url https://download.pytorch.org/whl/cpu
         rm -rf build
         
     - name: Pip build (with CUDA)
       if: ${{ matrix.enable_cuda }}
       run: |
         conda activate nnpops
+        sed -i -e '/"torch/c "torch==${{ matrix.pytorch }}",' pyproject.toml
         pip -vvv wheel --wheel-dir=$(pwd) .
         rm -rf build
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,6 +4,7 @@ requires = [
     "wheel",
     "ninja",
     "cmake>=3.23",
+    "torch>=2.0",
     "GitPython"
 ]
 build-backend = "setuptools.build_meta"