from mamba_ssm import Mamba

wurenkai / UltraLight-VM-UNet

[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".

234 stars 33 forks source link

from mamba_ssm import Mamba #10

Open 1245179435 opened 7 months ago

1245179435 commented 7 months ago

请问这个文件在哪里。。

wurenkai commented 7 months ago

Hi, you can find it in the official mamba code.

Lm0611 commented 7 months ago

ImportError: /public/home/ncu_418000230030/anaconda3/envs/Uvmunet/lib/python3.8/site-packages/causal_conv1d_cuda.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda20CUDACachingAllocator9allocatorE

请问这个问题怎么解决