JinmiaoChenLab/SEDR

CUDA out of memory

Closed this issue · 1 comments

I analysed Stereo-seq data by using SEDR, and an error was raised:

CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 11.73 GiB total capacity; 11.09 GiB already allocated; 1.19 MiB free; 11.51 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

Could anyone help me how to fix it?

I am sorry that I am testing SEDR on a GPU with 48GB. Probably you should try it on a larger GPU.