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)

相关推荐
程序员爱钓鱼14 分钟前
Python编程实战 · 基础入门篇 | 什么是Python
后端·python
程序猿小D15 分钟前
【完整源码+数据集+部署教程】遥感图像道路检测分割系统源码和数据集:改进yolo11-CARAFE
python·yolo·计算机视觉·目标跟踪·数据集·yolo11·遥感图像道路检测分割系统
zy_destiny22 分钟前
【工业场景】用YOLOv8实现人员打电话识别
人工智能·python·深度学习·yolo·机器学习·计算机视觉·目标跟踪
FreeBuf_22 分钟前
Happy DOM曝CVSS 9.4严重RCE漏洞,PoC已公开(CVE-2025-61927)
java·c语言·c++·python·php
C嘎嘎嵌入式开发34 分钟前
(10)100天python从入门到拿捏《Python中的数据结构与自定义数据结构》
数据结构·python·算法
程序员杰哥36 分钟前
Pytest与Unittest测试框架对比
自动化测试·软件测试·python·测试工具·测试用例·excel·pytest
wxin_VXbishe1 小时前
基于SpringBoot的天天商城管理系统的设计与现-计算机毕业设计源码79506
java·c++·spring boot·python·spring·django·php
睿思达DBA_WGX1 小时前
使用 python-docx 库操作 word 文档(3):读取word文档的内容
python·word
这里有鱼汤1 小时前
别再凭感觉画股票箱体了!用DBSCAN让你的策略更稳、更准、更客观
后端·python
AALoveTouch1 小时前
同程旅行签到脚本
python