Sobolev_INRs icon indicating copy to clipboard operation
Sobolev_INRs copied to clipboard

return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass

Open XiangFeng66 opened this issue 2 years ago • 1 comments

when I run this code ,I have this question Traceback (most recent call last): File "D:\软件\pycharm专业版\PyCharm 2022.1.3\plugins\python\helpers\pydev\pydevd.py", line 1491, in exec pydev_imports.execfile(file, globals, locals) # execute the script File "D:\软件\pycharm专业版\PyCharm 2022.1.3\plugins\python\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "D:/科研/Soblev_INrs/Sobolev_INRs-main/Experiments/inverse_rendering/train.py", line 696, in train() File "D:/科研/Soblev_INrs/Sobolev_INRs-main/Experiments/inverse_rendering/train.py", line 613, in train der_loss = der_mse(rgb, coordinate_s, target_grad_s) File "D:\科研\Soblev_INrs\Sobolev_INRs-main\Experiments\inverse_rendering\loss.py", line 15, in der_mse pred_grad_r = diff_operators.gradient( File "D:\科研\Soblev_INrs\Sobolev_INRs-main\Experiments\inverse_rendering\diff_operators.py", line 7, in gradient grad = torch.autograd.grad(y, [x], grad_outputs=grad_outputs, create_graph=True)[0] File "D:\Anaconda\envs\fengxiangNerf\lib\site-packages\torch\autograd_init.py", line 276, in grad return Variable.execution_engine.run_backward( # Calls into the C++ engine to run the backward pass RuntimeError: CUDA out of memory. Tried to allocate 64.00 MiB (GPU 0; 6.00 GiB total capacity; 5.17 GiB already allocated; 0 bytes free; 5.31 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split

I don't know the reason why this code have question

XiangFeng66 avatar Oct 17 '22 00:10 XiangFeng66

Sorry for the late reply. The reason for this RuntimeError is that your GPU memory is not large enough, you can try to run this code on GPU with larger memory.

wtyuan96 avatar Nov 13 '22 14:11 wtyuan96