causal_conv1d mamba_ssm

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

causal_conv1d mamba_ssm #6

Open sunkunjie opened 4 months ago

sunkunjie commented 4 months ago

pip install causal_conv1d==1.0.0
pip install mamba_ssm==1.0.1 Hello, I want to ask under the windows system, how to download this package, thank you very much

hqzqaq commented 4 months ago

In ubuntu, my version is:

mamba-ssm==1.1.4
causal-conv1d==1.1.3.post1
cuda==12.2

I can successfully run

JCruan519 commented 4 months ago

@sunkunjie Hello, currently the causal_conv1d and mamba_ssm libraries do not support the Windows platform. We recommend using the Linux platform, or staying updated on the following repositories: https://github.com/Dao-AILab/causal-conv1d/releases https://github.com/state-spaces/mamba/releases