Skip to content
forked from pyg-team/pyg-lib

Low-Level Graph Neural Network Operators for PyG

License

Notifications You must be signed in to change notification settings

cfgfung/pyg-lib

 
 

Repository files navigation

pyg-lib

Testing Status Docs Status Code Coverage

Installation

We provide pre-built Python wheels for all major OS/PyTorch/CUDA combinations from Python 3.7 till 3.11, 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

  • ${TORCH} should be replaced by either 1.11.0, 1.12.0, 1.13.0 or 2.0.0
  • ${CUDA} should be replaced by either cpu, cu102, cu113, cu115, cu116, cu117 or cu118

The following combinations are supported:

PyTorch 2.0 cpu cu102 cu113 cu115 cu116 cu117 cu118
Linux
Windows
macOS
PyTorch 1.13 cpu cu102 cu113 cu115 cu116 cu117 cu118
Linux
Windows
macOS
PyTorch 1.12 cpu cu102 cu113 cu115 cu116 cu117 cu118
Linux
Windows
macOS
PyTorch 1.11 cpu cu102 cu113 cu115 cu116 cu117 cu118
Linux
Windows
macOS

Form nightly

Nightly wheels are provided for Linux from Python 3.7 till 3.11:

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

From master

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

About

Low-Level Graph Neural Network Operators for PyG

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 62.0%
  • Python 26.0%
  • Cuda 9.6%
  • CMake 2.2%
  • C 0.2%