GPU out of memory

[hasan-ghavidel]

i received this error:

/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/functional.py:512: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3587.)
return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]
Best model: TransMorph_Validation_dsc0.744.pth.tar
Traceback (most recent call last):
File "/scratch/h/H.Ghavidel/TransMorph_Transformer_for_Medical_Image_Registration/IXI/TransMorph/infer_TransMorph.py", line 120, in
main()
File "/scratch/h/H.Ghavidel/TransMorph_Transformer_for_Medical_Image_Registration/IXI/TransMorph/infer_TransMorph.py", line 68, in main
x_def, flow = model(x_in)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "/scratch/h/H.Ghavidel/TransMorph_Transformer_for_Medical_Image_Registration/IXI/TransMorph/models/TransMorph.py", line 859, in forward
f5 = self.c2(x_s0)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/container.py", line 217, in forward
input = module(input)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
return forward_call(*args, **kwargs)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/conv.py", line 610, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/home/h/H.Ghavidel/venv/my_venv/lib64/python3.9/site-packages/torch/nn/modules/conv.py", line 605, in _conv_forward
return F.conv3d(
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 420.00 MiB. GPU

may anyone helps me to fix it