ERROR: Failed building wheel for causal-conv1d with torch>2.4

[we11as22]

pip install torch==2.4.0 helps

[xtwigs]

@tridao would it be possible to release torch 2.5 wheels?

[kzlinguistics]

it ended up being an NVCC version for me. I was able to solve it by doing the following:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

nvcc --version

[s-smits]

The main problem appears to be compatibility issues with causal-conv1d. There are two key issues:
Missing Precompiled Wheel: The installer tried to download a precompiled wheel for your specific configuration (Python 3.11 + CUDA 12.4 + torch 2.4.1) but got a 404 error, meaning this specific combination doesn't have a pre-built package.

Would the main problem be Python 3.11 (3.10 better) or PyTorch 2.4.1 (downgrading to 2.3.1)? Assuming CUDA 12.4 is fully backwards compatible with 12.1