Mamba安装

1.在windows的WSL上安装mamba

2.安装miniconda,

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
sudo chmod 777 Miniconda3-latest-Linux-x86_64.sh 
./Miniconda3-latest-Linux-x86_64.sh

3.打开激活

vim ~/.bashrc
export PATH=$PATH:/root/miniconda3/bin
source ~/miniconda3/bin/activate

我是在root下安装的miniconda

4.按以下进行安装

conda create -n HFppg python=3.9
conda activate your_env_name
conda install cudatoolkit==11.7 -c nvidia
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
conda install -c "nvidia/label/cuda-11.7.0" cuda-nvcc
conda install packaging
pip install importlib-metadata==4.13.0
pip install regex==2023.5.2
pip install click
pip install causal-conv1d==1.1.0
pip install mamba-ssm

5.测试

python
import torch
from mamba_ssm import Mamba