search

hustvl / Vim

Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
Apache License 2.0
2.55k stars 159 forks source link

Share the environment that worked. #67

Open uxhao-o opened 2 months ago

uxhao-o commented 2 months ago

I have been successfully run. Environment follows: cuda 11.8 python 3.10.13 pytorch 2.1.1 causal_conv1d 1.1.1 mamba-ssm 1.2.0.post1

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
pip install causal_conv1d==1.1.1
pip install mamba-ssm==1.2.0.post1
# copy mamba-ssm dir in vim to conda env site-package dir
cp -rf mamba-1p1p1/mamba_ssm /opt/miniconda3/envs/mamba/lib/python3.10/site-packages
macnack commented 2 months ago

it also works for me, thanks

ZXXaaaa commented 2 months ago

Hello, my environment is the same as yours, but I still report an error: Mamba.init() got an unexpected keyword argument 'bimamba'. Do you know how to solve it?

uxhao-o commented 2 months ago

Hello, my environment is the same as yours, but I still report an error: Mamba.init() got an unexpected keyword argument 'bimamba'. Do you know how to solve it?

This issue may be you didn't copy the mamba-ssm dir of Vim to the envs site-packages dir. Because Vim's author modified the mamba-ssm in their code. The mamba-ssm package that installed via pip is official package. So, you must copy mamba-ssm dir of Vim to virtual environment after pip install mamba-ssm.

cp -rf mamba-1p1p1/mamba_ssm /opt/miniconda3/envs/**mamba**/lib/python3.10/site-packages
ZXXaaaa commented 2 months ago

Hello, my environment is the same as yours, but I still report an error: Mamba.init() got an unexpected keyword argument 'bimamba'. Do you know how to solve it?

This issue may be you didn't copy the mamba-ssm dir of Vim to the envs site-packages dir. Because Vim's author modified the mamba-ssm in their code. The mamba-ssm package that installed via pip is official package. So, you must copy mamba-ssm dir of Vim to virtual environment after pip install mamba-ssm.

cp -rf mamba-1p1p1/mamba_ssm /opt/miniconda3/envs/**mamba**/lib/python3.10/site-packages

Thanks a lot. I know what the problem is

lxaw commented 2 months ago

This also worked for me, thank you!

abc-132 commented 1 month ago

Hello, in my environment “causal_conv1d_cuda“ can't be imported ,do you encount the same problem?

wwzh2626 commented 1 month ago

Hello, in my environment “causal_conv1d_cuda“ can't be imported ,do you encount the same problem?您好，在我的环境中“causal_conv1d_cuda”无法导入，您遇到同样的问题吗？

This problem might caused by the wrong version of causal_conv1d mamba-ssm. Follow this issue, import exact version of the packages as causal_conv1d==1.1.1 and mamba-ssm==1.2.0.post1!