准备
- 一个Linux系统,Change Mamba目前无法在Windows系统中运行
- 下载Change Mamba的代码(上一篇文章末尾附带链接)
配置pycharm环境
- 根据MambaCD代码中的README_zh-CN.md
- 点开Terminal创建虚拟环境(提示:有的Mamba对python版本有要求)
conda create --name changemamba python=3.10 conda activate changemamba
- 这时候如果执行成功,则继续跟着自带README教程走
pip install -r /yourPath/requirements.txt cd kernels/selective_scan && pip install .
报错1:requirements.txt
- 报超时错的话,自己单独下载torch(因为我的torch总是报错)
- 执行下列命令
nvcc -V
- 可以查看自己cuda版本(提示:有的mamba对cuda版本有要求)
- PyTorch官网找到合适自己的版本,我是11.7版本
- 复制语句去Terminal执行。我在这里并没报错,如果报错超时,可以多下载几次。
- 执行下列命令
pip list
可以看到自己刚下的torch版本 :
- 此时已经快成功了。执行下列命令
python
import torch
没报错的话,按住Ctrl+D退出python编译环境
- 此时可以继续执行README中的流程,报错后再来本文。
pip install -r /yourPath/requirements.txt
cd kernels/selective_scan && pip install .
pip install mmengine==0.10.1 mmcv==2.1.0 opencv-python-headless ftfy regex
pip install mmdet==3.3.0 mmsegmentation==1.2.2 mmpretrain==1.2.0
报错2:mmengine
- 安装依赖库时,总是报错超时,这里只能换网或者多下载几次。可能是我同时在下载预训练权重的原因,所以这里暂时不要同时执行多个下载任务。
多试几次后就步入正轨了:
报错3:预训练权重
- 注意:在下载预训练权重时,需要翻墙!!
以上是本人所踩的坑,希望对您有帮助~