Skip to content

Commit 38289bf

Browse files
authored
Fix CUDA wheels (part 2) (#488)
* PyTorch 2.7 support * PyTorch 2.7 support
1 parent cbf094d commit 38289bf

File tree

1 file changed

+5
-13
lines changed

1 file changed

+5
-13
lines changed

.github/workflows/building.yml

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ jobs:
4646
- name: Upgrade pip
4747
run: |
4848
pip install --upgrade setuptools
49+
pip install wheel
4950
5051
- name: Free Disk Space (Ubuntu)
5152
if: ${{ runner.os == 'Linux' }}
@@ -82,30 +83,21 @@ jobs:
8283
shell:
8384
bash
8485

85-
- name: Install main package for CPU
86+
- name: Build wheel for CPU
8687
if: ${{ matrix.cuda-version == 'cpu' }}
8788
run: |
88-
FORCE_ONLY_CPU=1 pip install -v -e . --no-build-isolation
89+
FORCE_ONLY_CPU=1 python setup.py bdist_wheel --dist-dir=dist
8990
shell:
9091
bash
9192

92-
- name: Install main package for GPU
93+
- name: Build wheel for GPU
9394
if: ${{ matrix.cuda-version != 'cpu' }}
9495
run: |
9596
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
96-
FORCE_CUDA=1 pip install -v -e . --no-build-isolation
97+
FORCE_CUDA=1 python setup.py bdist_wheel --dist-dir=dist
9798
shell:
9899
bash
99100

100-
- name: Test installation
101-
run: |
102-
python -c "import torch_scatter; print('torch-scatter:', torch_scatter.__version__)"
103-
104-
- name: Build wheel
105-
run: |
106-
pip install wheel
107-
python setup.py bdist_wheel --dist-dir=dist
108-
109101
- name: Configure AWS
110102
uses: aws-actions/configure-aws-credentials@v1
111103
with:

0 commit comments

Comments
 (0)