安装mamba_ssm和causal-conv1d报错 python setup.py egg_info did not run successfully和RuntimeError: mamba_ssm is only supported on CUDA 11.6 and above错误解决

问题

虚拟conda环境中使用如下命令安装pytorch
pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
然后安装mamba：pip install mamba_ssm
报错

这里表示cuda错误，但是安装的cuda就是cu117。

解决

mamba安装时是匹配系统默认的cuda，经过nvcc -V查看我的版本为cuda11.3。
因此，更改系统默认的cuda版本。
查看系统中安装的版本

# 查看系统中安装的版本
ls -l /usr/local | grep cuda
#当前使用版本
/usr/local/cuda/bin/nvcc --version

将11.3更换为更换11.7.
(1)方式一

# 终端输入
vim ~/.bashrc

# 切换到输入模型，在任意行插入如下字段：
export PATH=//usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64$LD_LIBRARY_PATH

# 保存，激活环境
source ~/.bashrc

(2)方式二
类似方式一，在bashrc插入的字段改为：

export CUDA_HOME='/usr/local/cuda-11.7'

然后再pip安装mamba_ssm即可。
注意：如还是安装不了，退出终端窗口再进去即可。