前期环境配置：

conda create -n vim python=3.10
conda activate vim
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging
pip install causal_conv1d-1.1.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install mamba_ssm-1.1.1-cp310-cp310-linux_x86_64.whl

具体请参考博客： https://blog.csdn.net/yyywxk/article/details/145005885

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

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