search

hgyhungry / ge-spmm

MIT License
101 stars 10 forks source link

out of memory #11

Open GugaGugaGuga opened 2 years ago

GugaGugaGuga commented 2 years ago

"RuntimeError: /pytorch/torch/csrc/autograd/profiler_cuda.cpp:23: out of memory" happened when I ran "python gcn_dgl.py --gpu=0 --dataset=pubmed --n-hidden=128 --n-layers=1" . The first run is successful and the rerun the second time has this problem.I seek for many solutions,but it has not yet been resolved.Highlight, I haven't changed this code.

hgyhungry commented 2 years ago

The script does not have non-deterministic code, and observations should not be different run-to-run. What you observed possibly signals a GPU memory leak, but I verified that this script causes no memory leak on my machine. You may want to check if it is the same on your machine, i.e. the first run leaves any device memory consumption on your GPU after the process exits. On my machine it leaves no memory un-freed, no matter I run how many times image