Installation

Pre-requistes

Pytorch-FINUFFT requires either finufft and/or cufinufft 2.2.0 or greater.

Note that currently, this version of cufinufft is unreleased and can only be installed from source. See the relevant installation pages for :external+finufft:doc:`finufft <install>` and :external+finufft:doc:`cufinufft <install_gpu>`.

Pytorch-FINUFFT also requires pytorch 2.0 or greater. See the installation matrix on Pytorch's website.

Once the pre-requisites are installed, you can install Pytorch-FINUFFT from source by running

pip install -U git+https://github.com/flatironinstitute/pytorch-finufft.git

Pytorch-FINUFFT is available on PyPI and can be installed with

pip install pytorch-finufft

which will also try to install compatible versions of torch and finufft.

You can also run

pip install pytorch-finufft[cuda]

to additionally install cufinufft.