报错的截图

在这里插入图片描述自认为报错的原因在于没有配置好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文件依然安装成功了。
在这里插入图片描述
在这里插入图片描述
心头大患终于解决了,没想到自己能够解决它!!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐