Home

Awesome

pyg-lib

Python Testing Status CPP Testing Status Docs Status Code Coverage

Installation

We provide pre-built Python wheels for all major OS/PyTorch/CUDA combinations from Python 3.9 till 3.12, see here. Note that currently, Windows wheels are not supported (we are working on fixing this as soon as possible).

To install the wheels, simply run

pip install pyg-lib -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html

where

The following combinations are supported:

PyTorch 2.4cpucu113cu116cu117cu118cu121cu124
Linux
Windows
macOS
PyTorch 2.3cpucu113cu116cu117cu118cu121cu124
Linux
Windows
macOS
PyTorch 2.2cpucu113cu116cu117cu118cu121cu124
Linux
Windows
macOS
                                                                               c
PyTorch 2.1cpucu113cu116cu117cu118cu121cu124
Linux
Windows
macOS
PyTorch 2.0cpucu113cu116cu117cu118cu121cu124
Linux
Windows
macOS
PyTorch 1.13cpucu113cu116cu117cu118cu121cu124
Linux
Windows
macOS
                                                                               c
PyTorch 1.12cpucu113cu116cu117cu118cu121cu124
Linux
Windows
macOS

Form nightly

Nightly wheels are provided for Linux from Python 3.9 till 3.12:

pip install pyg-lib -f https://data.pyg.org/whl/nightly/torch-${TORCH}+${CUDA}.html

From master

pip install ninja wheel
pip install git+https://github.com/pyg-team/pyg-lib.git