biomed-AI / ConGI

MIT License
10 stars 4 forks source link

Hardware Requirements #1

Open tqwang743 opened 1 year ago

tqwang743 commented 1 year ago

Hi, It is a fantastic method for analyzing spatial transcriptomics! I tried to train ConGI on a workstation with a GPU (RTX A4000) with 16.9 GB VRAM. But I encountered a problem named"cuda out of memory", so can you share the hardware requirements when training the ConGI Thank you very much!

kakaisi commented 4 months ago

hey,bro! Yes, this is a big problem, can you tell me about the hardware you are using, when I train on the server, I use the GPU in parallel, and I also reduce the training batch by a factor of 10, but it is still not enough