Linux服务器Mamba2安装及example运行问题

最新推荐文章于 2025-04-14 20:06:55 发布

我有一个大秃头

最新推荐文章于 2025-04-14 20:06:55 发布

阅读量3.8k

点赞数 29

文章标签：人工智能深度学习 python conda pip

本文链接：https://blog.csdn.net/weixin_53868994/article/details/140121763

版权

Mamba 从发表到现在已经有段时间了，先前的Mamba代码有些地方不完善或者需要对源码做些修改后编译，最近可能需要用到Mamba，发现之前实现的Vision Mamba块，注释掉了一些源码可能会导致训练速度下降，然后官方实现了Mamba2，这里尝试安装，做一些记录，防止服务器重置🫤

CUDA 安装

在安装其它东西之前，需要安装CUDA，这里参照了 Vision Mamba 的环境安装 cuda-11.8 版本，

直接到 CUDA 官网找对应版本就行 CUDA官网

在服务器上运行这两行代码，然后安装CUDA，网上有很多教程我这里服务器已经装好了，不在赘述

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run

Conda 创建环境

Anaconda 创建一个虚拟环境配置mamba，创建并激活，同样，这里创建python 3.10 的环境

conda create -n your_env_name python=3.10.13
conda activate your_env_name

Pytorch 安装

pytorch 这里安装2.1.1 版本，直接进入官网去找相应命令就行，PyTorch官网

最低0.47元/天解锁文章