pyg-team · akihironitta · Aug 29, 2025 · Aug 25, 2025 · Aug 26, 2025 · Aug 26, 2025
diff --git a/.github/actions/setup/action.yml b/.github/actions/setup/action.yml
@@ -39,7 +39,11 @@ runs:
 
     - name: Install PyTorch ${{ inputs.torch-version }}+${{ inputs.cuda-version }}
       run: |
-        pip install torch==${{ inputs.torch-version }} --extra-index-url https://download.pytorch.org/whl/${{ inputs.cuda-version }}
+        if [ ${{ inputs.torch-version }} == '2.9.0' ]; then
+          pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/${{ inputs.cuda-version }}
+        else
+          pip install torch==${{ inputs.torch-version }} --index-url https://download.pytorch.org/whl/${{ inputs.cuda-version }}
+        fi
         python -c "import torch; print('PyTorch:', torch.__version__)"
         python -c "import torch; print('CUDA:', torch.version.cuda)"
         python -c "import torch; print('CXX11 ABI:', torch.compiled_with_cxx11_abi())"

diff --git a/.github/workflows/cuda/Linux-env.sh b/.github/workflows/cuda/Linux-env.sh
@@ -1,6 +1,11 @@
 #!/bin/bash
 
 case ${1} in
+  cu130)
+    export FORCE_CUDA=1
+    export PATH=/usr/local/cuda-13.0/bin:${PATH}
+    export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;9.0;10.0;12.0+PTX"
+    ;;
   cu129)
     export FORCE_CUDA=1
     export PATH=/usr/local/cuda-12.9/bin:${PATH}

diff --git a/.github/workflows/cuda/Linux.sh b/.github/workflows/cuda/Linux.sh
@@ -4,6 +4,12 @@ OS=ubuntu2204
 
 # https://developer.nvidia.com/cuda-toolkit-archive
 case ${1} in
+  cu130)
+    CUDA=13.0
+    APT_KEY=${OS}-${CUDA/./-}-local
+    FILENAME=cuda-repo-${APT_KEY}_${CUDA}.0-580.65.06-1_amd64.deb
+    URL=https://developer.download.nvidia.com/compute/cuda/${CUDA}.1/local_installers
+    ;;
   cu129)
     CUDA=12.9
     APT_KEY=${OS}-${CUDA/./-}-local
@@ -81,7 +87,7 @@ sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600
 wget -nv ${URL}/${FILENAME}
 sudo dpkg -i ${FILENAME}
 
-if [ "${1}" = "cu117" ] || [ "${1}" = "cu118" ] || [ "${1}" = "cu121" ] || [ "${1}" = "cu124" ] || [ "${1}" = "cu126" ] || [ "${1}" = "cu128" ] || [ "${1}" = "cu129" ]; then
+if [ "${1}" = "cu117" ] || [ "${1}" = "cu118" ] || [ "${1}" = "cu121" ] || [ "${1}" = "cu124" ] || [ "${1}" = "cu126" ] || [ "${1}" = "cu128" ] || [ "${1}" = "cu129" ] || [ "${1}" = "cu130" ]; then
   sudo cp /var/cuda-repo-${APT_KEY}/cuda-*-keyring.gpg /usr/share/keyrings/
 else
   sudo apt-key add /var/cuda-repo-${APT_KEY}/7fa2af80.pub

diff --git a/.github/workflows/cuda/Windows-env.sh b/.github/workflows/cuda/Windows-env.sh
@@ -1,6 +1,10 @@
 #!/bin/bash
 
 case ${1} in
+  cu130)
+    export FORCE_CUDA=1
+    export PATH=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v13.0/bin:${PATH}
+    ;;
   cu129)
     export FORCE_CUDA=1
     export PATH=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v12.9/bin:${PATH}

diff --git a/.github/workflows/cuda/Windows.sh b/.github/workflows/cuda/Windows.sh
@@ -1,6 +1,11 @@
 #!/bin/bash
 
 case ${1} in
+  cu130)
+    CUDA_SHORT=13.0
+    CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.0/local_installers
+    CUDA_FILE=cuda_${CUDA_SHORT}.0_windows.exe
+    ;;
   cu129)
     CUDA_SHORT=12.9
     CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.1/local_installers

diff --git a/.github/workflows/utils/build_linux.sh b/.github/workflows/utils/build_linux.sh
@@ -13,9 +13,14 @@ echo "FORCE_CUDA: ${FORCE_CUDA}"
 echo "TORCH_CUDA_ARCH_LIST: ${TORCH_CUDA_ARCH_LIST}"
 
 export CIBW_BUILD="cp${PYTHON_VERSION//./}-manylinux_x86_64"
-export CIBW_BEFORE_BUILD="pip install ninja wheel setuptools && pip install torch==${TORCH_VERSION} --index-url https://download.pytorch.org/whl/${CUDA_VERSION}"
 # pyg-lib doesn't have torch as a dependency, so we need to explicitly install it when running tests.
-export CIBW_BEFORE_TEST="pip install pytest && pip install torch==${TORCH_VERSION} --index-url https://download.pytorch.org/whl/${CUDA_VERSION}"
+if [[ "${TORCH_VERSION}" == "2.9.0" ]]; then
+  export CIBW_BEFORE_BUILD="pip install ninja wheel setuptools && pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/${CUDA_VERSION}"
+  export CIBW_BEFORE_TEST="pip install pytest && pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/${CUDA_VERSION}"
+else
+  export CIBW_BEFORE_BUILD="pip install ninja wheel setuptools && pip install torch==${TORCH_VERSION} --index-url https://download.pytorch.org/whl/${CUDA_VERSION}"
+  export CIBW_BEFORE_TEST="pip install pytest && pip install torch==${TORCH_VERSION} --index-url https://download.pytorch.org/whl/${CUDA_VERSION}"
+fi
 
 if [[ "${CUDA_VERSION}" == "cu"* ]]; then
   # Use CUDA-pre-installed image

diff --git a/.github/workflows/utils/full_matrix_linux.json b/.github/workflows/utils/full_matrix_linux.json
@@ -1,4 +1,9 @@
 [
+  {
+    "torch-version": "2.9.0",
+    "python-version": ["3.10", "3.11", "3.12", "3.13"],
+    "cuda-version": ["cpu", "cu130"]
+  },
   {
     "torch-version": "2.8.0",
     "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],

diff --git a/.github/workflows/utils/full_matrix_macos.json b/.github/workflows/utils/full_matrix_macos.json
@@ -1,4 +1,14 @@
 [
+  {
+    "torch-version": "2.9.0",
+    "python-version": ["3.10", "3.11", "3.12", "3.13"],
+    "cuda-version": ["cpu"]
+  },
+  {
+    "torch-version": "2.8.0",
+    "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],
+    "cuda-version": ["cpu"]
+  },
   {
     "torch-version": "2.7.0",
     "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],

diff --git a/.github/workflows/utils/full_matrix_windows.json b/.github/workflows/utils/full_matrix_windows.json
@@ -1,4 +1,9 @@
 [
+  {
+    "torch-version": "2.9.0",
+    "python-version": ["3.10", "3.11", "3.12", "3.13"],
+    "cuda-version": ["cpu"]
+  },
   {
     "torch-version": "2.8.0",
     "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],

diff --git a/.github/workflows/utils/minimal_matrix_linux.json b/.github/workflows/utils/minimal_matrix_linux.json
@@ -1,4 +1,9 @@
 [
+  {
+    "torch-version": "2.9.0",
+    "python-version": ["3.10", "3.13"],
+    "cuda-version": ["cu130"]
+  },
   {
     "torch-version": "2.8.0",
     "python-version": ["3.9", "3.13"],

diff --git a/.github/workflows/utils/minimal_matrix_macos.json b/.github/workflows/utils/minimal_matrix_macos.json
@@ -1,6 +1,11 @@
 [
   {
-    "torch-version": "2.7.0",
+    "torch-version": "2.9.0",
+    "python-version": ["3.10", "3.13"],
+    "cuda-version": ["cpu"]
+  },
+  {
+    "torch-version": "2.8.0",
     "python-version": ["3.9", "3.13"],
     "cuda-version": ["cpu"]
   },

diff --git a/.github/workflows/utils/minimal_matrix_windows.json b/.github/workflows/utils/minimal_matrix_windows.json
@@ -1,4 +1,9 @@
 [
+  {
+    "torch-version": "2.9.0",
+    "python-version": ["3.10", "3.13"],
+    "cuda-version": ["cpu"]
+  },
   {
     "torch-version": "2.8.0",
     "python-version": ["3.9", "3.13"],

diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,7 +7,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 ### Added
 - Added support for `manylinux_2_28` wheels built with CUDA in nightly releases ([#496](https://github.com/pyg-team/pyg-lib/pull/496))
 - Added support for `manylinux_2_28` wheels in nightly releases ([#480](https://github.com/pyg-team/pyg-lib/pull/480))
-- Added CUDA 13.0 support ([#509](https://github.com/pyg-team/pyg-lib/pull/509))
+- Added CUDA 13.0 support ([#509](https://github.com/pyg-team/pyg-lib/pull/509), [#511](https://github.com/pyg-team/pyg-lib/pull/511))
 - Added CUDA 12.9 support ([#494](https://github.com/pyg-team/pyg-lib/pull/494))
 - Added PyTorch 2.8 support ([#494](https://github.com/pyg-team/pyg-lib/pull/494))
 - Added PyTorch 2.7 support ([#442](https://github.com/pyg-team/pyg-lib/pull/442))