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
407 stars 13 forks source link

cuda version #7

Closed 2962313771 closed 4 months ago

2962313771 commented 4 months ago

image 这里有两个cuda版本，应该选择哪个呢，不能同时有两个吧

2962313771 commented 4 months ago

我在执行下面的句子的时候出现了很多错误，能不能推荐一个cuda版本

pip install causal_conv1d==1.0.0 # causal_conv1d-1.0.0+cu118torch1.13cxx11abiFALSE-cp38-cp38-linux_x86_64.whl pip install mamba_ssm==1.0.1 # mmamba_ssm-1.0.1+cu118torch1.13cxx11abiFALSE-cp38-cp38-linux_x86_64.whl

ZhiQiangYang513 commented 4 months ago

在多次安装后，我认为cuda11.8是最低要求。

hqzqaq commented 4 months ago

在ubuntu中，我的版本为：

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

可以成功运行

JCruan519 commented 4 months ago

@2962313771 You need to configure CUDA 11.8 in your environment to install and use the causal_conv1d and mamba_ssm libraries. For Torch-related libraries, simply follow the instructions mentioned above for installation.

2962313771 commented 4 months ago

OK,Thanks.

niu-yq commented 3 months ago

功

你好，我想问一下你的torch是哪个版本的

hqzqaq commented 3 months ago

我的torch是2.2.1的版本

niu-yq commented 3 months ago

你好，我在pytorch官网上看到的目前cuda只有12.1的版本，torch版本也有对应是2.2.1的，但是当我把cuda改成12.2就安装失败了，而torch2.2.1和cuda12.1又会出现这个报错，另外我还尝试了torch 2.0.0+cu118也是会有这个报错，可以问一下怎么下载torch12.2+cu122吗（我的mamba-ssm==1.2.0.post1 causal-conv1d==1.1.3.post1）

。。 @.***

 

------------------ 原始邮件 ------------------ 发件人: "JCruan519/VM-UNet" @.>; 发送时间: 2024年4月4日(星期四) 晚上7:52 @.>; @.**@.>; 主题: Re: [JCruan519/VM-UNet] cuda version (Issue #7)

我的torch是2.2.1的版本

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

hqzqaq commented 3 months ago

尝试一下这个命令吧

pip3 install torch torchvision torchaudio

我忘记我咋装的了，当时我装环境也卡了一下，后面很快解决了，忘记咋弄的了

niu-yq commented 3 months ago

好的，谢谢你！！！

。。 @.***

 

------------------ 原始邮件 ------------------ 发件人: "JCruan519/VM-UNet" @.>; 发送时间: 2024年4月4日(星期四) 晚上10:03 @.>; @.**@.>; 主题: Re: [JCruan519/VM-UNet] cuda version (Issue #7)

尝试一下这个命令吧 pip3 install torch torchvision torchaudio

我忘记我咋装的了，当时我装环境也卡了一下，后面很快解决了，忘记咋弄的了

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>