Skip to content

Commit 3d1d9e3

Browse files
authored
Fix CUDA wheels (part 2) (#255)
1 parent 281f8b1 commit 3d1d9e3

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: Install scipy
5152
if: ${{ (matrix.python-version == '3.8') || (matrix.python-version == '3.9') }}
@@ -79,30 +80,21 @@ jobs:
7980
shell:
8081
bash
8182

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

89-
- name: Install main package for GPU
90+
- name: Build wheel for GPU
9091
if: ${{ matrix.cuda-version != 'cpu' }}
9192
run: |
9293
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
93-
FORCE_CUDA=1 pip install -v -e . --no-build-isolation
94+
FORCE_CUDA=1 python setup.py bdist_wheel --dist-dir=dist
9495
shell:
9596
bash
9697

97-
- name: Test installation
98-
run: |
99-
python -c "import torch_cluster; print('torch-cluster:', torch_cluster.__version__)"
100-
101-
- name: Build wheel
102-
run: |
103-
pip install wheel
104-
python setup.py bdist_wheel --dist-dir=dist
105-
10698
- name: Configure AWS
10799
uses: aws-actions/configure-aws-credentials@v1
108100
with:

0 commit comments

Comments
 (0)