(cuda12.1)selective_scan-0.0.2-cp310-cp310-linux_x86_64.whl

作品简介

在Linux下安装 Vmamba(不带core)

前期环境配置:

conda create -n vmamba python=3.10
conda activate vmamba
conda install cuda-nvcc==12.1.105
# 或者 conda install nvidia/label/cuda-12.1.0::cuda-nvcc
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121
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。

创作时间: