Ubuntu24.04.1系统下VideoMamba环境配置

文章目录


前言

VideoMamba环境的配置折磨了我三天,由于Mamba对Cuda的版本有要求,因此配置环境的时候Cuda版本以及各种包的版本问题是最需要注意的,下面我来一步一步的演示如何配好VideoMamba模型的运行环境。

第一步:基本的环境创建

创建python版本为3.10.13的conda环境,在这里我将环境命名为videomamba

bash 复制代码
conda create -n videomamba python=3.10.13

激活创建的环境。

bash 复制代码
conda activate videomamba

然后安装版本为11.8CUDA

bash 复制代码
conda install cudatoolkit==11.8 -c nvidia

安装版本2.1.1pytorch

bash 复制代码
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

安装版本为11.8cuda-nvcc库。

bash 复制代码
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc

第二步:causal-conv1d和mamba_ssm库的安装

大多数的教程都是直接pip安装这两个库,但是直接安装的方式基本不可能安装成功,所以采用离线安装的方式来装。

bash 复制代码
pip install causal-conv1d
pip install mamba_ssm

首先下载用于离线安装causal-conv1dwhl文件,下载链接: causal-conv1d

在这里我下载的是causal_conv1d-1.0.0+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

然后下载用于离线安装mamba_ssmwhl文件,下载链接: mamba_ssm,在这里,我下载的是mamba_ssm-1.0.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

然后安装两个whl

c 复制代码
pip install causal_conv1d-1.0.0+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install mamba_ssm-1.0.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

注意 在这里安装的mamba_ssm的环境运行videomamba代码是会报错的,所以我们需要把videomamba代码中的mamba文件夹下的mamba_ssm文件夹替换到我们配置的python环境中,具体如下:

首先我们去anaconda3/envs/videomamba/lib/python3.10/site-packages/找到mamba_ssm文件夹,然后删除它

接着,我们将VideoMamba-main/mamba下的mamba_ssm文件复制到anaconda3/envs/videomamba/lib/python3.10/site-packages/,这样我们的环境就配好了,其中VideoMamba-main是videomamba的github上的项目名。

第三步:安装requirements.txt

这里两个特别需要注意的两个包:apex==0.1xformers==0.0.24

对于apex包,我们直接使用pip安装会失败,所以也需要离线安装,下载链接: apex

然后手动安装

bash 复制代码
python setup.py install

对于xformers==0.0.24包,官方给的requirements.txt文件中的版本是0.0.24,但是安装0.0.24的包会自动将我们安装的torch包更新为2.2版本的,导致配的环境出现问题,因此我们需要安装0.0.23版本的包,这样就不会更新torch包了。

将原始的requirements.txt文件中的xformers==0.0.24改为xformers==0.0.23,然后直接安装就行了。

bash 复制代码
pip install -r requirements.txt

参考:

1\] [Mamba环境安装指南:包括causal_conv1d及mamba-ssm的安装教程](https://www.iotword.com/26572.html) \[2\] [复现Vision Mamba时报错"TypeError: causal_conv1d_fwd(): incompatible function arguments."](https://blog.csdn.net/Qanyh/article/details/138812171?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-138812171-blog-138437338.235%5Ev43%5Econtrol&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-138812171-blog-138437338.235%5Ev43%5Econtrol&utm_relevant_index=2) \[3\] [史上最强Mamba环境配置教程](https://blog.csdn.net/qq_45645368/article/details/140777871) \[4\] [最详细的mamba讲解及环境配置(包含踩坑心得)](https://blog.csdn.net/m0_63745825/article/details/138136975?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ECtr-1-138136975-blog-140777871.235%5Ev43%5Epc_blog_bottom_relevance_base7&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ECtr-1-138136975-blog-140777871.235%5Ev43%5Epc_blog_bottom_relevance_base7&utm_relevant_index=2) \[5\] [VideoMamba](https://zhuanlan.zhihu.com/p/692570798) \[6\] [VideoMamba 开源项目教程](https://blog.csdn.net/gitblog_00148/article/details/141239139) \[7\] [vision-mamba环境配置](https://zhuanlan.zhihu.com/p/697931372) \[8\] [Linux服务器上安装 Vision Mamba 虚拟环境---全面教程](https://zhuanlan.zhihu.com/p/687359086) \[9\] [Mamba环境配置](https://zhuanlan.zhihu.com/p/691962837) \[10\] [如何在ubuntu配置Vision-Mamba环境](https://blog.csdn.net/m0_51323948/article/details/138248245)

相关推荐
databook2 小时前
Manim实现闪光轨迹特效
后端·python·动效
Juchecar3 小时前
解惑:NumPy 中 ndarray.ndim 到底是什么?
python
用户8356290780513 小时前
Python 删除 Excel 工作表中的空白行列
后端·python
Json_3 小时前
使用python-fastApi框架开发一个学校宿舍管理系统-前后端分离项目
后端·python·fastapi
数据智能老司机10 小时前
精通 Python 设计模式——分布式系统模式
python·设计模式·架构
数据智能老司机11 小时前
精通 Python 设计模式——并发与异步模式
python·设计模式·编程语言
数据智能老司机11 小时前
精通 Python 设计模式——测试模式
python·设计模式·架构
数据智能老司机11 小时前
精通 Python 设计模式——性能模式
python·设计模式·架构
c8i11 小时前
drf初步梳理
python·django
每日AI新事件11 小时前
python的异步函数
python