GitHub

CUDA error:out of memory

I have a question how to avoid "RuntimeError: CUDA error:out of memory" during the IS2RS task. The IS2RS task was carried out by the code below python main.py --mode ...
GitHub

[torch_memory_saver.cpp] CUresult error: 2 (out of memory) file=csrc/utils.h func=cu_mem_create line=187

Error Context: Training via ray.get(actor_model.async_train(...)) Error Log:[torch_memory_saver.cpp] CUresult error: 2 (out of memory) file=csrc/utils.h func=cu_mem ...