(cuda12.8)Linux下 Vim (Vision Mamba)环境安装包

作品简介

Linux 下安装Vim,适合算力 12.0。

前期环境准备

在base环境安装cuda12.8,详细步骤参考博客:https://blog.csdn.net/yyywxk/article/details/151864570

conda create -n mamba python=3.10
conda activate mamba


# CUDA 12.8
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128
python -c "import torch; print(torch.cuda.is_available())" # 验证torch安装

安装

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

pip install causal_conv1d-1.4.0-cp310-cp310-linux_x86_64.whl
pip install mamba_ssm-1.1.1-cp310-cp310-linux_x86_64.whl --no-dependencies causal_conv1d

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

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

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

安装有问题加CSDN博客vx。

创作时间: