当前位置: 代码网 > 服务器>服务器>Linux > Ubuntu20.04下成功安装Mamba的运行环境

Ubuntu20.04下成功安装Mamba的运行环境

2024年07月31日 Linux 我要评论
conda环境内配置Mamba运行环境。
  • 参考链接:
  1. mamba 环境安装踩坑问题汇总及解决方法 
  2. [最佳实践] conda环境内安装cuda 和 mamba的安装 

我的环境参数:

安装步骤:

  • 使用conda命令创建新的虚拟环境,这里的我的your_env_name是设置的python_3.10。
conda create -n your_env_name python=3.10.13
conda activate your_env_name
conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia

接下来按照参考链接1和2中的步骤进行安装,下面的安装皆在激活的虚拟环境python_3.10中安装。

conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging

接下来安装causal-conv1d:

git clone https://github.com/dao-ailab/causal-conv1d.git 
cd causal-conv1d 
git checkout v1.2.0 # current latest version tag 
causal_conv1d_force_build=true pip install .

这里安装的时候,可能会出现一些错误,比如我出现的错误是:

error: could not build wheels for causal-conv1d, which is required to install pyproject.toml-based projects
  • 解决方案1:这个时候不要仅仅盯着这个错误,要看这个错误的上面一部分是报什么错误,优先解决那个错误,然后再次按照causal-conv1d的步骤再走一遍,不过这里克隆过的文件夹已经存在,可以不用再次进行克隆。
  • 解决方案2:我还找到一个原因是,科学上网的原因,先关掉梯子,试一下上述步骤,看看是什么报错,然后再打开梯子,再看一下是什么报错,还有就是,可以打开全局模式,再试一下。
  • 接下来,安装mamba-ssm。
cd ..
git clone https://github.com/state-spaces/mamba.git
cd ./mamba
git checkout v1.2.0 # current latest version tag
mamba_force_build=true pip install .

        安装这一个步骤的时候我的报错是:

building wheel for mamba-ssm (setup.py) ... error error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully. 
│ exit code: 1 ╰─> [8 lines of output]

code
  torch.__version__  = 2.1.1+cu118
  
  
  running bdist_wheel
  guessing wheel url:  https://github.com/state-spaces/mamba/releases/download/v1.2.0.post1/mamba_ssm-1.2.0.post1+cu122torch2.1cxx11abifalse-cp310-cp310-linux_x86_64.whl
  error: <urlopen error [errno 110] connection timed out>
  [end of output]
note: this error originates from a subprocess, and is likely not a problem with pip. error: failed building wheel for mamba-ssm running setup.py clean for mamba-ssm failed to build mamba-ssm error: could not build wheels for mamba-ssm, which is required to install pyproject.toml-based projects.

解决方案是我下载了上述中提到的这个whl文件,然后手动安装了这个文件。下载这个文件的链接:http:// https://github.com/state spaces/mamba/releases/download/v1.2.0.post1/mamba_ssm-1.2.0.post1+cu122torch2.1cxx11abifalse-cp310-cp310-linux_x86_64.whl

  • 手动安装后,再次执行安装mamba-ssm的步骤。
  •  我的虚拟环境中所有的安装包展示:

  • 测试安装是否成功: 

这里也有说是gcc版本的问题,在参考链接2中,说到gcc版本要在9.0以上,我的版本是11.2,所以没有更换版本,如果还出现报错的话,可以尝试更换gcc版本再次进行安装。

注:如果内容有误,请留言指导,感谢!

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com