服务器部署UMamba分割MRI：causal_conv1d和mamba_ssm离线下载（pip不可用）、修改batch_size和epoch

UMamba官方仓库：bowang-lab/U-Mamba: U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation

My实验配置：

为了避免后续麻烦，尽量我的配置：cu118 虚拟环境安装python 3.10、torch2.4及更老版本，否则无对应安装包，且会报

in <module> import selective_scan_cuda ImportError: .../selective_scan_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: ...

1.mamba_ssm库离线安装

step1 检索正确版本的安装包并下载到本地

Releases · state-spaces/mamba

使用wget指令在linux服务器控制台下载离线安装包

wget https://github.com/state-spaces/mamba/releases/download/v2.2.0/mamba_ssm-2.2.0+cu118torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

根据你的设备配置选择合适的下载链接，以上为cuda版本11.8，torch版本2.4，python3.10，abi=false的指令。使用如下代码检查abi选true/false：

import torch
print(torch._C._GLIBCXX_USE_CXX11_ABI)

step2 cd进入包安装路径，使用pip install将包拖拽到光标处，回车

2.causal-conv1d库离线安装

step1 检索正确版本的安装包并下载

Releases · Dao-AILab/causal-conv1d

wget https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.5.4/causal_conv1d-1.5.4+cu11torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

step2 同上

3.UMamba修改epoch、batchsize

 修改batchsize→  .../nnUNet_preprocessed/Dataset999_OAIZIB/nnUNetPlans.json

修改epoch轮数→ .../U-Mamba/umamba/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py

4. 最后就可克隆U-Mamba并使用和nnUNet类似的指令训练分割模型了

注意有Enc和Bot两种trainer，我选择Bot（bottleneck？）