teddykoker / torchsort

Fast, differentiable sorting and ranking in PyTorch
https://pypi.org/project/torchsort/
Apache License 2.0
765 stars 33 forks source link

How to build the CUDA extension? #42

Closed xfcxing closed 2 years ago

teddykoker commented 2 years ago

Hi, the CUDA extension should build automatically with pip install torchsort if you have the CUDA toolkit (nvcc) installed.

You can run pip install --force-reinstall --no-cache-dir torchsort to force a rebuild of the extension if necessary.