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

GitHub 加速计划 / ma / mamba
8
2
下载
最近提交(Master分支:4 个月前 )
62db608d 4 个月前
323db269 Co-authored-by: Tri Dao <tridao@users.noreply.github.com> 4 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐