当前位置:   article > 正文

VideoMamba

videomamba

太夸张了!!!

环境安装:

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

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的项目路径下

https://link.zhihu.com/?target=https%3A//github.com/Dao-AILab/causal-conv1d/releases/download/v1.1.3/causal_conv1d-1.1.3%2Bcu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

https://link.zhihu.com/?target=https%3A//github.com/Dao-AILab/causal-conv1d/archive/refs/tags/v1.1.3.zip

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

  1. cd Videomamba-main/causal-conv1d-1.1.3
  2. pip install .
  3. cd Videomamba-main/mamba-1.1.1
  4. 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文件夹替换进去)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/秋刀鱼在做梦/article/detail/878396
推荐阅读
相关标签
  

闽ICP备14008679号