diff --git a/.github/workflows/cuda/cu129-Windows.sh b/.github/workflows/cuda/cu129-Windows.sh index c6764fec..24bfeb4d 100755 --- a/.github/workflows/cuda/cu129-Windows.sh +++ b/.github/workflows/cuda/cu129-Windows.sh @@ -6,8 +6,8 @@ curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq 7z x "/tmp/gpu_driver_dlls.zip" -o"/c/Windows/System32" export CUDA_SHORT=12.9 -export CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.0/local_installers -export CUDA_FILE=cuda_${CUDA_SHORT}.0_571.96_windows.exe +export CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.1/local_installers +export CUDA_FILE=cuda_${CUDA_SHORT}.1_576.57_windows.exe # Install CUDA: curl -k -L "${CUDA_URL}/${CUDA_FILE}" --output "${CUDA_FILE}" diff --git a/README.md b/README.md index f397b671..f0b1b9c0 100644 --- a/README.md +++ b/README.md @@ -45,6 +45,22 @@ All included operations are broadcastable, work on varying data types, are imple We provide pip wheels for all major OS/PyTorch/CUDA combinations, see [here](https://data.pyg.org/whl). +#### PyTorch 2.8 + +To install the binaries for PyTorch 2.8.0, simply run + +``` +pip install torch-scatter -f https://data.pyg.org/whl/torch-2.8.0+${CUDA}.html +``` + +where `${CUDA}` should be replaced by either `cpu`, `cu126`, `cu128`, or `cu129` depending on your PyTorch installation. + +| | `cpu` | `cu126` | `cu128` | `cu129` | +|-------------|-------|---------|---------|---------| +| **Linux** | ✅ | ✅ | ✅ | ✅ | +| **Windows** | ✅ | ✅ | ✅ | ✅ | +| **macOS** | ✅ | | | | + #### PyTorch 2.7 To install the binaries for PyTorch 2.7.0, simply run