超详细无踩坑mamba环境配置
适用于Ubuntu20.4,cuda11.6
1. 创建mamba环境
// python版本安装了3.10.13
conda create -n mamba python==3.10.13
conda activate mamba
2. 安装pytorch,去官网按照cuda版本11.6安装了torch
// torch版本安装了1.13.1
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
3. 安装必要的包
pip install packing
4. 安装causal_conv1d
pip install causal_conv1d==1.1.1
5. 安装mamba_ssm(请注意和causal_conv1d版本一致)
pip install mamba_ssm==1.1.1