在Windows下安装 Vmamba(带core),终极版,支持算力为12.0 等目前常见GPU。
Windows 下前期环境准备
在base环境安装cuda12.8,详细步骤参加博客。
conda create -n mamba python=3.10
conda activate mamba
# CUDA 12.8
pip install torch-2.8.0.dev20250323+cu128-cp310-cp310-win_amd64.whl
pip install torchvision-0.22.0.dev20250324+cu128-cp310-cp310-win_amd64.whl
pip install torchaudio-2.6.0.dev20250324+cu128-cp310-cp310-win_amd64.whl # 可选
python -c "import torch; print(torch.cuda.is_available())" # 验证torch安装triton-windows 环境准备
https://blog.csdn.net/yyywxk/article/details/145018635
https://blog.csdn.net/yyywxk/article/details/144868136
安装
下载后在这个文件目录下执行:
pip install selective-scan-0.0.2-cp310-cp310-win_amd64.whl
具体请参考本人CSDN博客及其系列博客:https://blog.csdn.net/yyywxk/article/details/146798627
注意 python3.10,torch2.8以及cuda12.8版本一定要与博客一致。
安装有问题加CSDN博客vx。