VideoMamba

Urdanny

已于 2024-04-15 15:04:46 修改

阅读量355

点赞数 5

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

于 2024-04-11 22:48:59 首次发布

本文链接：https://blog.csdn.net/Urdanny/article/details/137656481

版权

太夸张了！！！

这次的项目遇到好多问题。我按照踩坑的顺序罗列下来。

1. 从最基础的底层开始关注！！！看下显卡驱动版本、gcc版本这些(gcc>=9)。看了太多博文分享的内容，都是直接给命令，很容易出错，而且明明是一步步按照要求的，老是报一些莫名其妙的错误。

2. 暂时只支持Linux系统，未在Windows下测试。

好了！开始！（默许上述满足要求）

1. 创建环境

conda create -n your_env_name python=3.10.13

conda activate your_env_name

conda install cudatoolkit==11.8 -c nvidia

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/wh

conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc

conda install packaging

至此需要注意的是Mamba要求CUDA>=11.6, 在你的虚拟环境里安装一个CUDA11.8版本比较合适。

下面这部分参考：https://zhuanlan.zhihu.com/p/687359086这位的内容。

访问这两个链接去下载casual-conv1d-1.1.3的包和source code并放在Videomamba-main的项目路径下

进入新加的两个source code里，安装对应的环境

cd Videomamba-main/causal-conv1d-1.1.3

pip install .

cd Videomamba-main/mamba-1.1.1

pip install .

别急！

首先找到你所创建的环境里的mamba_ssm文件夹，路径一般在这里："/envs/mamba/lib/python3.10/site-packages/mamba_ssm/"，将其替换为你项目代码里的mamba_ssm文件夹：

rm -rf "/envs/mamba/lib/python3.10/site-packages/mamba_ssm/" (删除原环境里的mamba_ssm)

cp -r "Videomamba-main/mamba-1p1p1/mamba_ssm" "/envs/mamba/lib/python3.10/site-packages/" (将项目code里的mamba_ssm文件夹替换进去)

关注