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官网
评论前必须登录!
注册