【人工智能】Linux系统Mamba安装流程

SmallBambooCode

已于 2024-09-22 14:03:22 修改

阅读量150

点赞数 2

文章标签： linux 服务器机器学习深度学习人工智能 python 神经网络

于 2024-09-22 13:59:07 首次发布

本文链接：https://blog.csdn.net/qq_35516428/article/details/142436172

版权

在编译安装 mamba 之前，你需要确保已安装正常的PyTorch环境。

# 安装必要的系统依赖
sudo apt update
sudo apt install build-essential
# 安装mamba依赖
pip install packaging wheel
# 克隆仓库
git clone https://github.com/Dao-AILab/causal-conv1d.git
git clone https://github.com/state-spaces/mamba.git
# 安装causal-conv1d
cd causal-conv1d
git checkout v1.2.0 # 调整到要安装的版本
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .
# 安装mamba
cd ../mamba
git checkout v1.2.0 # 调整到要安装的版本
MAMBA_FORCE_BUILD=TRUE pip install .