Skip to content

Commit 09150c4

Browse files
rusty1srusty1s
committed
update
1 parent cdfbc7e commit 09150c4

6 files changed

Lines changed: 55 additions & 47 deletions

File tree

‎.github/workflows/building.yml‎

Lines changed: 8 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -15,42 +15,24 @@ jobs:
1515
fail-fast: false
1616
matrix:
1717
os: [ubuntu-22.04, macos-14, windows-2022, ubuntu-22.04-arm]
18-
python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
19-
torch-version: [2.8.0]
20-
cuda-version: ['cpu', 'cu118', 'cu124', 'cu126', 'cu128', 'cu129']
18+
python-version: ['3.10', '3.11', '3.12', '3.13', '3.14']
19+
torch-version: [2.9.0, 2.10.0]
20+
cuda-version: ['cpu', 'cu126', 'cu128', 'cu130']
2121
exclude:
22-
- torch-version: 2.6.0
23-
cuda-version: 'cu128'
24-
- torch-version: 2.6.0
25-
cuda-version: 'cu129'
26-
- torch-version: 2.7.0
27-
cuda-version: 'cu124'
28-
- torch-version: 2.7.0
29-
cuda-version: 'cu129'
30-
- torch-version: 2.8.0
31-
cuda-version: 'cu118'
32-
- torch-version: 2.8.0
33-
cuda-version: 'cu124'
34-
- os: macos-14
35-
cuda-version: 'cu118'
36-
- os: macos-14
37-
cuda-version: 'cu124'
22+
- os: windows-2022
23+
cuda-version: 'cu130'
3824
- os: macos-14
3925
cuda-version: 'cu126'
4026
- os: macos-14
4127
cuda-version: 'cu128'
4228
- os: macos-14
43-
cuda-version: 'cu129'
44-
- os: ubuntu-22.04-arm
45-
cuda-version: 'cu118'
46-
- os: ubuntu-22.04-arm
47-
cuda-version: 'cu124'
29+
cuda-version: 'cu130'
4830
- os: ubuntu-22.04-arm
4931
cuda-version: 'cu126'
5032
- os: ubuntu-22.04-arm
5133
cuda-version: 'cu128'
5234
- os: ubuntu-22.04-arm
53-
cuda-version: 'cu129'
35+
cuda-version: 'cu130'
5436

5537
steps:
5638
- uses: actions/checkout@v4
@@ -68,9 +50,8 @@ jobs:
6850
pip install wheel
6951
7052
- name: Install scipy
71-
if: ${{ matrix.python-version == '3.9' }}
7253
run: |
73-
pip install scipy==1.10.1
54+
pip install scipy
7455
7556
- name: Free Disk Space (Ubuntu)
7657
if: ${{ runner.os == 'Linux' }}

‎.github/workflows/cuda/cu130-Linux-env.sh‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
#!/bin/bash
2+
3+
CUDA_HOME=/usr/local/cuda-13.0
4+
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
5+
PATH=${CUDA_HOME}/bin:${PATH}
6+
7+
export FORCE_CUDA=1
8+
export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;9.0;10.0;12.0+PTX"

‎.github/workflows/cuda/cu130-Linux.sh‎

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
#!/bin/bash
2+
3+
OS=ubuntu2204
4+
5+
wget -nv https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin
6+
sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600
7+
8+
wget -nv https://developer.download.nvidia.com/compute/cuda/13.0.2/local_installers/cuda-repo-${OS}-13-0-local_13.0.2-580.95.05-1_amd64.deb
9+
10+
sudo dpkg -i cuda-repo-${OS}-13-0-local_13.0.2-580.95.05-1_amd64.deb
11+
sudo cp /var/cuda-repo-${OS}-13-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
12+
13+
sudo apt-get -qq update
14+
sudo apt install cuda-nvcc-13-0 cuda-libraries-dev-13-0
15+
sudo apt clean
16+
17+
rm -f *.deb

‎.github/workflows/testing.yml‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ jobs:
1919
fail-fast: false
2020
matrix:
2121
os: [ubuntu-latest, windows-2022]
22-
python-version: [3.9]
23-
torch-version: [2.7.0, 2.8.0]
22+
python-version: ['3.10']
23+
torch-version: [2.9.0, 2.10.0]
2424

2525
steps:
2626
- uses: actions/checkout@v4
@@ -53,9 +53,8 @@ jobs:
5353
echo WITH_METIS=0 >> $GITHUB_ENV
5454
5555
- name: Install scipy
56-
if: ${{ matrix.python-version == '3.9' }}
5756
run: |
58-
pip install scipy==1.10.1
57+
pip install scipy
5958
6059
- name: Install main package
6160
run: |

‎README.md‎

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -34,55 +34,55 @@ Note that only `value` comes with autograd support, as `index` is discrete and t
3434

3535
We provide pip wheels for all major OS/PyTorch/CUDA combinations, see [here](https://data.pyg.org/whl).
3636

37-
#### PyTorch 2.8
37+
#### PyTorch 2.10
3838

39-
To install the binaries for PyTorch 2.8.0, simply run
39+
To install the binaries for PyTorch 2.10, simply run
4040

4141
```
42-
pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.8.0+${CUDA}.html
42+
pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.10.0+${CUDA}.html
4343
```
4444

45-
where `${CUDA}` should be replaced by either `cpu`, `cu126`, `cu128`, or `cu129` depending on your PyTorch installation.
45+
where `${CUDA}` should be replaced by either `cpu`, `cu126`, `cu128`, or `cu130` depending on your PyTorch installation.
4646

47-
| | `cpu` | `cu126` | `cu128` | `cu129` |
47+
| | `cpu` | `cu126` | `cu128` | `cu130` |
4848
|-------------|-------|---------|---------|---------|
4949
| **Linux** | ✅ | ✅ | ✅ | ✅ |
5050
| **Windows** | ✅ | ✅ | ✅ | ✅ |
5151
| **macOS** | ✅ | | | |
5252

53-
#### PyTorch 2.7
53+
#### PyTorch 2.9
5454

55-
To install the binaries for PyTorch 2.7.0, simply run
55+
To install the binaries for PyTorch 2.9, simply run
5656

5757
```
58-
pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.7.0+${CUDA}.html
58+
pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.9.0+${CUDA}.html
5959
```
6060

61-
where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu126`, or `cu128` depending on your PyTorch installation.
61+
where `${CUDA}` should be replaced by either `cpu`, `cu126`, `cu128`, or `cu130` depending on your PyTorch installation.
6262

63-
| | `cpu` | `cu118` | `cu126` | `cu128` |
63+
| | `cpu` | `cu126` | `cu128` | `cu130` |
6464
|-------------|-------|---------|---------|---------|
6565
| **Linux** | ✅ | ✅ | ✅ | ✅ |
6666
| **Windows** | ✅ | ✅ | ✅ | ✅ |
6767
| **macOS** | ✅ | | | |
6868

69-
#### PyTorch 2.6
69+
#### PyTorch 2.8
7070

71-
To install the binaries for PyTorch 2.6.0, simply run
71+
To install the binaries for PyTorch 2.8, simply run
7272

7373
```
74-
pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.6.0+${CUDA}.html
74+
pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.8.0+${CUDA}.html
7575
```
7676

77-
where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu124`, or `cu126` depending on your PyTorch installation.
77+
where `${CUDA}` should be replaced by either `cpu`, `cu126`, `cu128`, or `cu129` depending on your PyTorch installation.
7878

79-
| | `cpu` | `cu118` | `cu124` | `cu126` |
79+
| | `cpu` | `cu126` | `cu128` | `cu129` |
8080
|-------------|-------|---------|---------|---------|
8181
| **Linux** | ✅ | ✅ | ✅ | ✅ |
8282
| **Windows** | ✅ | ✅ | ✅ | ✅ |
8383
| **macOS** | ✅ | | | |
8484

85-
**Note:** Binaries of older versions are also provided for PyTorch 1.4.0, PyTorch 1.5.0, PyTorch 1.6.0, PyTorch 1.7.0/1.7.1, PyTorch 1.8.0/1.8.1, PyTorch 1.9.0, PyTorch 1.10.0/1.10.1/1.10.2, PyTorch 1.11.0, PyTorch 1.12.0/1.12.1, PyTorch 1.13.0/1.13.1, PyTorch 2.0.0/2.0.1, PyTorch 2.1.0/2.1.1/2.1.2, PyTorch 2.2.0/2.2.1/2.2.2, PyTorch 2.3.0/2.3.1, PyTorch 2.4.0/2.4.1, and PyTorch 2.5.0/2.5.1 (following the same procedure).
85+
**Note:** Binaries of older versions are also provided for PyTorch 1.4.0, PyTorch 1.5.0, PyTorch 1.6.0, PyTorch 1.7.0/1.7.1, PyTorch 1.8.0/1.8.1, PyTorch 1.9.0, PyTorch 1.10.0/1.10.1/1.10.2, PyTorch 1.11.0, PyTorch 1.12.0/1.12.1, PyTorch 1.13.0/1.13.1, PyTorch 2.0.0/2.0.1, PyTorch 2.1.0/2.1.1/2.1.2, PyTorch 2.2.0/2.2.1/2.2.2, PyTorch 2.3.0/2.3.1, PyTorch 2.4.0/2.4.1, PyTorch 2.5.0/2.5.1, PyTorch 2.6.0, and PyTorch 2.7.0/2.7.1 (following the same procedure).
8686
For older versions, you need to explicitly specify the latest supported version number or install via `pip install --no-index` in order to prevent a manual installation from source.
8787
You can look up the latest supported version number [here](https://data.pyg.org/whl).
8888

‎setup.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,9 @@ def get_extensions():
9292
nvcc_flags = os.getenv('NVCC_FLAGS', '')
9393
nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
9494
nvcc_flags += ['-O3']
95+
nvcc_flags += ['-DTORCH_INDUCTOR_CPP_WRAPPER']
96+
if sys.platform == 'win32':
97+
nvcc_flags += ['-Xcompiler', '/permissive-']
9598
if torch.version.hip:
9699
# USE_ROCM was added to later versions of PyTorch
97100
# Define here to support older PyTorch versions as well:

0 commit comments

Comments
 (0)