在Windows下安装 Vmamba(带core),终极版,兼容算力低于7.0高于5.0的老旧GPU,同时支持7.0-9.0 等目前常见GPU。
Windows 下前期环境准备
conda create -n mamba python=3.10
conda activate mamba
# CUDA 12.4
pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu124
python -c "import torch; print(torch.cuda.is_available())" # 验证torch安装
# 安装cuda
conda install nvidia/label/cuda-12.4.0::cuda-nvcc
pip install setuptools==68.2.2
conda install packaging
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-linux_x86_64.whl
具体请参考本人CSDN博客及其系列博客:https://blog.csdn.net/yyywxk/article/details/140422758
注意 python torch 及 cuda 版本一定要和博客一致。
安装有问题加CSDN博客vx。