最近Mamba作为Transformer的挑战者非常的火热,许多相关工作都不断涌现。但在复现的时候往往会卡在安装mamba-ssm这个库上。当尝试使用 pip install mamba-ssm 进行安装的时候往往会卡在下图位置 为了探究卡在了building的哪一步,加入--verbose进行显示 pip install mamba-ssm --no-cache-dir --verbose 发现其会从gith...
最正确的修复方式是找到setup.py,将其中的wheel地址换位能够正常访问的地址。但奈何没有时间查找pip下载的setup.py的位置,于是就直接用pip安装匹配到的wheel。 比如错误信息中出现 Guessing wheel URL: https://github.com/state-spaces/mamba/releases/download/v2.2.2/mamba_ssm-2.2.2+cu122torch2.4cxx11abiFALSE...
3、 受博文 “flash-attention踩坑:使用conda管理CUDA”启发,合理调整按照顺利,先安装CUDA,并且安装cuda-nvcc,正确的安装步骤如下: conda create -n your_env_name python=3.10.13conda activate your_env_nameconda install cudatoolkit==11.8 -c nvidiapip install torch==2.1.1 torchvision==0.16.1 torchaudio...
error: identifier “__cudaDeviceSynchronizeDeprecationAvoidance” is undefined ModuleNotFoundError: No module named ‘packaging’ 可以在安装mamba-ssm和causal_conv1d前先conda install packaging,具体参考文末链接[2] AttributeError: ‘str’ object has no attribute ‘contiguous’ 验证是否安装成功 如何知道自己...
1、使用网友配置好的Docker环境,参考:解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像DockHub仓库地址:https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general代码:docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1 ...