Issue with torchsort CUDA extension: ImportError and missing toolchain on Linux

[ilovesdu]

Hi there,

I encountered an error while using the torchsort package on Linux. The error message is as follows:

ImportError: You are trying to use the torchsort CUDA extension, but it looks like it is not available. Make sure you have the CUDA toolchain installed, and reinstall torchsort with pip install --force-reinstall --no-cache-dir torchsort to rebuild the extension.

It seems that the CUDA extension for torchsort is not available, and the system is unable to locate the required CUDA toolchain. I have already tried reinstalling the package using the suggested command, but the issue persists.

Here are the details of my environment:

Operating system: Linux
CUDA version: 11.8 (via nvidia-cuda-runtime-cu11)
PyTorch version: 2.4.1+cu118
Installed torchsort version: 0.1.9
Relevant NVIDIA libraries:
nvidia-cudnn-cu11 9.1.0.70
nvidia-cuda-runtime-cu11 11.8.89
nvidia-nvtx-cu11 11.8.86
Would appreciate any guidance on how to resolve this issue.

Thanks in advance!