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
471 stars 22 forks source link

当我改变gpu_ids=[1,2,3]时，报错（是否不支持多卡运行？） #20

Open ShChen233 opened 6 months ago

ShChen233 commented 6 months ago

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:2 and cuda:1! (when checking argument for argument mat2 in method wrapper_CUDA_bmm)

JCruan519 commented 6 months ago

@ShChen233 Hello, the current code does not support multi-GPU running. Additionally, training with a single GPU is sufficient.