(cuda12.8)Window下Mamba2环境安装包终极版

作品简介

在Windows下安装 mamba_ssm 2.2.2,终极版。

Windows 下前期环境准备

在base环境安装cuda12.8。

conda create -n mamba python=3.10
conda activate mamba

# CUDA 12.8
pip install --pre torch torchvision  --index-url https://download.pytorch.org/whl/nightly/cu128

triton-windows 环境准备

https://blog.csdn.net/yyywxk/article/details/145018635

https://blog.csdn.net/yyywxk/article/details/144868136

安装

下载后在这个文件目录下执行:

pip install causal_conv1d-1.4.0-cp310-cp310-win_amd64.whl
pip install mamba_ssm-2.2.2-cp310-cp310-win_amd64.whl

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

注意 python3.10,torch2.8以及cuda12.8版本一定要与博客一致。

支持算力为12.0 等目前常见GPU。

安装有问题加CSDN博客vx。

创作时间: