已从源码编译causal conv1d和mamba,亲测可用
版本:
causal-conv1d 1.4.0
mamba_ssm 2.2.2
cu11.8+python3.11和cu12.4+3.12测试没问题
1.cd进入文件夹
2.pip install .
通过百度网盘分享的文件:torch.zip
链接:https://pan.baidu.com/s/1OX_5bv4OBuep1PRE8JRoDw?pwd=ujoy
提取码:ujoy
- 缺少triton先pip install
- 缺少packaging先pip install
- selective_scan_cuda问题见windows系统下anaconda中配置Mamba官方代码环境