在编译安装 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 .