赞
踩
太夸张了!!!
这次的项目遇到好多问题。我按照踩坑的顺序罗列下来。
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文件夹替换进去)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。