(Vim cuda12.4)mamba_ssm-1.1.1-cp310-cp310-win_amd64.whl

作品简介

在Windows下安装 Vim 需要的mamba_ssm 1.1.1,终极版。

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 mamba_ssm-2.2.2-cp310-cp310-win_amd64.whl --no-dependencies causal_conv1d

具体请参考本人CSDN博客及其系列博客:https://blog.csdn.net/yyywxk/article/details/145018635

注意 python torch 及 cuda 版本一定要与博客一致。

支持算力为6.0-9.0 等目前常见GPU。

安装有问题加CSDN博客vx。

创作时间: