JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".
Apache License 2.0
407 stars 13 forks source link

Also meet the mamba_ssm error #13

Open mywebinfo65536 opened 4 months ago

mywebinfo65536 commented 4 months ago

Hi, I follow your instruction to install the mamba_ssm by using cuda11.8, linux system and the 2 whl files, and there is no error underling for the import clause in the code. but errors happen when runing the code:

image

image

Could you help to tell me the reason, thanks.

JCruan519 commented 4 months ago

@mywebinfo65536 Hello, if you followed the instructions provided completely, there shouldn't be any errors, as I've successfully installed it on different machines myself. I suggest you check if there are similar issues in the mamba_ssm or causal_conv1d repositories.

https://github.com/state-spaces/mamba https://github.com/Dao-AILab/causal-conv1d