CUDA out of memory on RTX 4090

[coconut0012]

Hi, @RickySkywalker !
I found that running drags operation in the demo will encounter the GPU memory problem with a 4090 GPU. Is there any way to reduce the GPU memory costs?

[ZHAOZHIHAO]

Hi @OctoNinja9 , have you managed to run this code?