mambaIR模型 pip install causal_conv1d==1.0.0时报错,已解决。
pip install causal_conv1d==1.0.0 mamba_ssm==1.0.1报错
·
报错的截图
自认为报错的原因在于没有配置好cuda与pytorch,今天又重新配置一遍,最终成功安装causal_conv1d1.0.0与mamba_ssm1.0.1。
一开始,下面四行代码分别输出False、2.0.1、None、None。
print(torch.cuda.is_available())
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
当我配置好cuda、cuDNN、pytorch后输出为:
有了以上的环境条件,我起初分别执行下面两行代码:
pip install causal-conv1d==1.0.0
pip install mamba_ssm==1.0.1
不再出现博客开头的报错,但是速度非常慢。于是我从下面两个链接中下载了如下图所示的.whl文件:
https://github.com/Dao-AILab/causal-conv1d/releases/tag/v1.0.0
https://github.com/state-spaces/mamba/releases/tag/v1.0.1

虽然我的cuda版本是11.7,但是使用cu118的.whl文件依然安装成功了。

心头大患终于解决了,没想到自己能够解决它!!
更多推荐


所有评论(0)