云计算百科
云计算领域专业知识百科平台

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

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

目录

CUDA 安装

Conda 创建环境

Pytorch 安装

causal-conv1d安装

Mamba 安装

问题解决


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)
未经允许不得转载:网硕互联帮助中心 » Linux服务器Mamba2安装及example运行问题
分享到: 更多 (0)

评论 抢沙发

评论前必须登录!