官方没有编译好的适用于Windows版本的whl,因此需要手动编译。笔者编译好了 Windows 下的 causalconv1d-1.1.1-cp310-cp310-winamd64.whl 和 mambassm-1.1.3-cp310-cp310-winamd64.whl ,可直接下载安装(只适用于torch 2.1)
下载后直接运行代码:
pip install causal_conv1d-1.1.1-cp310-cp310-win_amd64.whl
出现如下图即安装成功causal-conv1d,成功安装之后,会在相应虚拟环境中(xxx\conda\envs\xxx\Lib\site-packages\)产生 causal_conv1d_cuda.cp310-win_amd64.pyd 文件,此文件对应 causal_conv1d_cuda 包。

pip install mamba_ssm-1.1.3-cp310-cp310-win-amd64.whl
出现如下图即安装成功mamba-ssm
