RuntimeError: CUDA error: out of memory

[ReekiLee]

Hello！
Thanks for your great code! However, I met a bug. I prepared the enviroment and installed it according to your README file, after that I ran demo.py, and I met this:

Traceback (most recent call last):
File "demo.py", line 22, in
im1_v = im1.cuda()
RuntimeError: CUDA error: out of memory

I used nvidia-smi to check the GPU but it had enough memory. I don't know how to fix the bug. Could you give me some advice and ideas? Thank you in advance!

[Yuliang-Zou]

Did you the empty GPU or other occupied GPUs? Or did you load your custom high-resolution image?