CUDA memory error

[viX-shaw]

Can you help me, I am new to pytorch.

**Traceback (most recent call last):
File "SST/train_ua.py", line 246, in
train()
File "SST/train_ua.py", line 184, in train
out = net(img_pre, img_next, boxes_pre, boxes_next, valid_pre, valid_next)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 150, in forward
return self.module(*inputs[0], **kwargs[0])
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/content/SST/layer/sst.py", line 68, in forward
x_next = self.forward_vgg(x_next, self.vgg, sources_next)
File "/content/SST/layer/sst.py", line 179, in forward_vgg
x = vggk
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py", line 338, in forward
self.padding, self.dilation, self.groups)
RuntimeError: CUDA out of memory. Tried to allocate 1.55 GiB (GPU 0; 11.17 GiB total capacity; 9.28 GiB already allocated; 1.12 GiB free; 448.72 MiB cached)

[wangaixue]

@viX-shaw at least two 16G memory

[EddieEduardo]

@wangaixue OH, JESUS!!! Mine is just 4G , are there any good solutions making it compatible with PC？Thanks for your reply.