Building wheels for collected packages: causal-conv1d
Building wheel for causal-conv1d (setup.py) … error
error: subprocess-exited-with-error
× python setup.py bdist_wheel did not run successfully.
│ exit code: 1
╰─> [8 lines of output]
torch.__version__ = 2.0.0+cu118running bdist_wheelGuessing wheel URL: https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.2.0.post2/causal_conv1d-1.2.0.post2+cu118torch2.0cxx11abiFALSE-cp311-cp311-linux_x86_64.whlerror: Remote end closed connection without response[end of output]
note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for causal-conv1d
Running setup.py clean for causal-conv1d
Failed to build causal-conv1d
ERROR: Could not build wheels for causal-conv1d, which is required to install pyproject.toml-based projects
解决:
根据自己的CUDA、torch、python版本,选择自己对应 mamba_ssm 版本,下载后离线安装即可。
https://github.com/state-spaces/mamba/releases
causal-conv1d包下载地址
https://github.com/Dao-AILab/causal-conv1d/releases