search

MrBlankness / LightM-UNet

Pytorch implementation of "LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation"
https://arxiv.org/abs/2403.05246
Apache License 2.0
279 stars 25 forks source link

Is a 24G NVIDIA Geforce RTX 4090 GPU enough to run all the experiments? #29

Open Fym68 opened 3 months ago

Fym68 commented 3 months ago

How much video memory is needed to run this project with 3D data? I currently have NVIDIA Geforce RTX 4090 GPU with 24G video memory, but it says out of memory

ztgzttt commented 2 months ago

How much video memory do I need to run this project with 3D data? I currently have an NVIDIA Geforce RTX 24 GPU with 4090G of video memory, but it says it's running out of memory

Do you know why, please? I have the same problem.