在Linux下安装 Vmamba(带core)。
前期环境配置:
conda create -n vmamba python=3.10 conda activate vmamba conda install cuda-nvcc==12.1.105 pip install -r requirements.txt
下载后在这个文件目录下执行:
pip install selective_scan-0.0.2-cp310-cp310-linux_x86_64.whl
具体请参考本人CSDN博客及其系列博客:https://blog.csdn.net/yyywxk/article/details/140418043
注意 python torch 及 cuda 版本一定要一致。
支持算力为7.0-9.0 等目前常见GPU。
安装有问题加CSDN博客vx。