CUDA out of memory. Tried to allocate 130.00 MiB (GPU 0; 6.00 GiB total capacity; 4.83 GiB already allocated;

[jiangningqu70]

When I click "Drag it" button, it has an error as follow. How can I do? Thanks.

Traceback (most recent call last): File "D:\Soft\python\lib\site-packages\gradio\routes.py", line 406, in run_predict event_data=event_data, File "D:\Soft\python\lib\site-packages\gradio\blocks.py", line 1300, in process_api fn_index, inputs, iterator, request, event_id, event_data File "D:\Soft\python\lib\site-packages\gradio\blocks.py", line 1037, in call_function utils.async_iteration, iterator, limiter=self.limiter File "D:\Soft\python\lib\site-packages\anyio\to_thread.py", line 34, in run_sync func, args, cancellable=cancellable, limiter=limiter File "D:\Soft\python\lib\site-packages\anyio_backends_asyncio.py", line 877, in run_sync_in_worker_thread return await future File "D:\Soft\python\lib\site-packages\anyio_backends_asyncio.py", line 807, in run result = context.run(func, args) File "D:\Soft\python\lib\site-packages\gradio\utils.py", line 488, in async_iteration return next(iterator) File "E:/workspace/python_workspace/DragGAN-main/gradio_app.py", line 135, in on_drag max_iters=max_iters): File "E:\workspace\python_workspace\DragGAN-main\drag_gan.py", line 183, in drag_gan sample2, F2 = g_ema.generate(latent, noise) File "E:\workspace\python_workspace\DragGAN-main\drag_gan.py", line 107, in generate out = conv1(out, latent[:, i], noise=noise1) File "D:\Soft\python\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl return forward_call(*input, kwargs) File "E:\workspace\python_workspace\DragGAN-main\stylegan2\model.py", line 376, in forward out = self.conv(input, style) File "D:\Soft\python\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl return forward_call(*input, *kwargs) File "E:\workspace\python_workspace\DragGAN-main\stylegan2\model.py", line 297, in forward out = self.blur(out) File "D:\Soft\python\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl return forward_call(input, kwargs) File "E:\workspace\python_workspace\DragGAN-main\stylegan2\model.py", line 104, in forward out = upfirdn2d(input, self.kernel, pad=self.pad) File "E:\workspace\python_workspace\DragGAN-main\stylegan2\op\upfirdn2d.py", line 186, in upfirdn2d_native out = _upfirdn2d_native(input, kernel, up, down, *pad) File "E:\workspace\python_workspace\DragGAN-main\stylegan2\op\upfirdn2d.py", line 205, in _upfirdn2d_native out, [0, 0, max(pad_x0, 0), max(pad_x1, 0), max(pad_y0, 0), max(pad_y1, 0)] File "D:\Soft\python\lib\site-packages\torch\nn\functional.py", line 4153, in _pad return _VF.constant_pad_nd(input, pad, value) RuntimeError: CUDA out of memory. Tried to allocate 130.00 MiB (GPU 0; 6.00 GiB total capacity; 4.83 GiB already allocated; 0 bytes free; 5.17 GiB reserved in total by PyTorch)