search

JCruan519 / VM-UNet

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

”pip install mamba_ssm==1.0.1“and"pip install causal_conv1d==1.0.0" Report an error #34

Closed wakakkakak closed 3 months ago

wakakkakak commented 3 months ago

Hello, I would like to ask the following error occurs when installing "pip install mamba_ssm==1.0.1", and I wonder how to solve it? I was running on a linux server. When I type the command line: cuda When I type the command line: image When I type the command line: image

I wonder what the problem is? And how to fix it? I look forward to hearing from you!

JCruan519 commented 3 months ago

@wakakkakak Hello, your CUDA driver needs to be upgraded to at least version 11.8 in order to use mamba_ssm.

wakakkakak commented 3 months ago

@wakakkakak Hello, your CUDA driver needs to be upgraded to at least version 11.8 in order to use mamba_ssm.

Thank you for your replying! I