Installation

Requirement

Install PyTorch >= 1.9.0 according the official PyTorch instructions. Pytorch>=2.8.0 is required for Triton-based convolutions.

Installation via Pip

Run the following commands to install ocnn given PyTorch has been installed.

pip install ocnn

Installation from Source

It is also easy to install ocnn from source.

1. Clone the code:

   git clone https://github.com/octree-nn/ocnn-pytorch.git
   

2. Enter the folder containing the code, and run the following command:

   pip install .
   

Note

The Triton-based convolution operations in ocnn require PyTorch ≥ 2.8.0 and the triton package. On Ubuntu systems, the triton package is automatically installed during the PyTorch installation process. However, on Windows systems, the triton package is not installed automatically, so you need to install it manually using the following command: pip install triton-windows.