WIN11中安装Mamba常见问题解决方案

1. 安装triton，下面提供一个WIN10/11编译好的Triton版本：

triton-2.0.0-cp310-cp310-win_amd64.whl
运行指令：pip install triton-2.0.0-cp310-cp310-win_amd64.whl

2. 安装causal-conv1d

FORCE_BUILD = os.getenv("CAUSAL_CONV1D_FORCE_BUILD", "FALSE") == "TRUE"
SKIP_CUDA_BUILD = os.getenv("CAUSAL_CONV1D_SKIP_CUDA_BUILD", "FALSE") == "TRUE"
# For CI, we want the option to build with C++11 ABI since the nvcr images use C++11 ABI
FORCE_CXX11_ABI = os.getenv("CAUSAL_CONV1D_FORCE_CXX11_ABI", "FALSE") == "TRUE"

将三个TRUE改为FALSE:

FORCE_BUILD = os.getenv("CAUSAL_CONV1D_FORCE_BUILD", "FALSE") == "FALSE"
SKIP_CUDA_BUILD = os.getenv("CAUSAL_CONV1D_SKIP_CUDA_BUILD", "FALSE") == "FALSE"
# For CI, we want the option to build with C++11 ABI since the nvcr images use C++11 ABI
FORCE_CXX11_ABI = os.getenv("CAUSAL_CONV1D_FORCE_CXX11_ABI", "FALSE") == "FALSE"

 3. 问题3：RuntimeError: mamba_ssm is only supported on CUDA 11.6 and above.

直接升级升级到11.8： conda install cudatoolkit==11.8 -c nvidia
安装NVCC：conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc

或者官方下载手动安装：CUDA Toolkit 11.8 Downloads