Skip to content

Commit 3d10cdc

Browse files
authored
Document PyTorch 2.8 installation in README (#496)
* update * update
1 parent efbe465 commit 3d10cdc

File tree

2 files changed

+18
-2
lines changed

2 files changed

+18
-2
lines changed

.github/workflows/cuda/cu129-Windows.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq
66
7z x "/tmp/gpu_driver_dlls.zip" -o"/c/Windows/System32"
77

88
export CUDA_SHORT=12.9
9-
export CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.0/local_installers
10-
export CUDA_FILE=cuda_${CUDA_SHORT}.0_571.96_windows.exe
9+
export CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.1/local_installers
10+
export CUDA_FILE=cuda_${CUDA_SHORT}.1_576.57_windows.exe
1111

1212
# Install CUDA:
1313
curl -k -L "${CUDA_URL}/${CUDA_FILE}" --output "${CUDA_FILE}"

README.md

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,22 @@ All included operations are broadcastable, work on varying data types, are imple
4545

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

48+
#### PyTorch 2.8
49+
50+
To install the binaries for PyTorch 2.8.0, simply run
51+
52+
```
53+
pip install torch-scatter -f https://data.pyg.org/whl/torch-2.8.0+${CUDA}.html
54+
```
55+
56+
where `${CUDA}` should be replaced by either `cpu`, `cu126`, `cu128`, or `cu129` depending on your PyTorch installation.
57+
58+
| | `cpu` | `cu126` | `cu128` | `cu129` |
59+
|-------------|-------|---------|---------|---------|
60+
| **Linux** |||||
61+
| **Windows** |||||
62+
| **macOS** || | | |
63+
4864
#### PyTorch 2.7
4965

5066
To install the binaries for PyTorch 2.7.0, simply run

0 commit comments

Comments
 (0)