pytorch_cluster

(★ 928)

PyTorch Extension Library of Optimized Graph Cluster Algorithms

File Explorer

Download Latest Version (.zip)

- - building.yml
  - linting.yml
  - stale.yml
  - testing.yml
- TorchClusterConfig.cmake.in
- - fps_cpu.cpp
  - fps_cpu.h
  - graclus_cpu.cpp
  - graclus_cpu.h
  - grid_cpu.cpp
  - grid_cpu.h
  - knn_cpu.cpp
  - knn_cpu.h
  - radius_cpu.cpp
  - radius_cpu.h
  - rw_cpu.cpp
  - rw_cpu.h
  - sampler_cpu.cpp
  - sampler_cpu.h
  - utils.h
- - fps_cuda.cu
  - fps_cuda.h
  - graclus_cuda.cu
  - graclus_cuda.h
  - grid_cuda.cu
  - grid_cuda.h
  - knn_cuda.cu
  - knn_cuda.h
  - nearest_cuda.cu
  - nearest_cuda.h
  - radius_cuda.cu
  - radius_cuda.h
  - rw_cuda.cu
  - rw_cuda.h
  - utils.cuh
- cluster.h
- extensions.h
- fps.cpp
- graclus.cpp
- grid.cpp
- knn.cpp
- macros.h
- nearest.cpp
- radius.cpp
- rw.cpp
- sampler.cpp
- version.cpp
- test_fps.py
- test_graclus.py
- test_grid.py
- test_knn.py
- test_nearest.py
- test_radius.py
- test_rw.py
- test_sampler.py
- __init__.py
- fps.py
- graclus.py
- grid.py
- knn.py
- nearest.py
- radius.py
- rw.py
- sampler.py
- testing.py
- typing.py
.coveragerc
.gitignore
CMakeLists.txt
LICENSE
MANIFEST.in
pyproject.toml
README.md
setup.cfg
setup.py

# Installation Guide

1. Get the code

git clone https://github.com/rusty1s/pytorch_cluster

Downloads the entire project code from GitHub to your computer.

cd pytorch_cluster

Moves into the project folder you just downloaded.

2. Official Install Script

Easy Recommended

Prerequisites

Python 3 Python is required to use pip.

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

Installs the package published on PyPI directly — no need to clone the source.

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

Installs the package published on PyPI directly — no need to clone the source.

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

Installs the package published on PyPI directly — no need to clone the source.

pip install torch-cluster

Installs the package published on PyPI directly — no need to clone the source.

✅ After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.

Pulled directly from this repo's README.

3. CMake

Medium

Prerequisites

Git Needed to download the project code from GitHub.
CMake The tool used to generate build configuration.
C/C++ 컴파일러 Windows needs Visual Studio (Community edition, free), macOS needs Xcode Command Line Tools, Linux needs the gcc/g++ package.

cmake ..

Analyzes the source code and generates build configuration files (must be run inside the build folder).

✅ Check that an executable was created inside the build folder, then run it directly (e.g. ./build/app_name).

Pulled directly from this repo's README.

4. Python

Easy

Prerequisites

Git Needed to download the project code from GitHub.
Python 3 On Windows, be sure to check 'Add Python to PATH' during installation.

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

Installs the package published on PyPI directly — no need to clone the source.

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

Installs the package published on PyPI directly — no need to clone the source.

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

Installs the package published on PyPI directly — no need to clone the source.

pip install torch-cluster

Installs the package published on PyPI directly — no need to clone the source.

✅ If it runs without errors and prints output in the terminal, it worked.

Pulled directly from this repo's README.

pytorch_cluster

File Explorer

# Installation Guide

2. Official Install Script

3. CMake

4. Python

Today's picks

Was this content helpful?

Recent Feedback

pytorch_cluster

File Explorer

# Installation Guide

2. Official Install Script

3. CMake

4. Python

Today's picks

Was this content helpful?

Recent Feedback

Download README