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

Starlight Spark 2024-06-27 17:37:03 阅读 84

参考链接:

Mamba 环境安装踩坑问题汇总及解决方法https://blog.csdn.net/yyywxk/article/details/136071016 

[最佳实践] conda环境内安装cuda 和 Mamba的安装 https://blog.csdn.net/weixin_44007713/article/details/136475398

我的环境参数:

安装步骤:

使用conda命令创建新的虚拟环境,这里的我的your_env_name是设置的python_3.10。

conda create -n your_env_name python=3.10.13

conda activate your_env_name

根据CUDA Version选择需要安装的pytorch版本。 链接:https://pytorch.org/get-started/locally/在激活的环境中,安装pytorch。

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版本再次进行安装

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



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。