vmamba 尝试安装记录

1、安装虚拟环境
conda create -n vmamba python==3.10
2、激活环境
conda activate vmamba
3、安装附加环境
pip install -r requirements.txt
4、确保cuda为11.7以上
5、安装selective_scan函数

下载原版,vmamba中的 selective_scan函数:

原版链接
https://github.com/MzeroMiko/VMambaicon-default.png?t=O83Ahttps://github.com/MzeroMiko/VMamba

上面这个文件夹在那个路径,就该路径执行下面的语句就行


cd kernels/selective_scan && pip install .
6、安装原版的causal_conv1d,mamba_ssm
pip install causal_conv1d mamba_ssm

tip:如果提示triton版本有问题,那么请安装低版本的triton
triton==2.1.0

你可能感兴趣的:(conda,python)