Skip to content

How did you test on multi gpu? #4

@vnyk-dgrid

Description

@vnyk-dgrid

I am trying to test the model and getting out of memory issue, were you able to et results using Single GPU?

Add frame 89 , 92 frames in total
Add frame 90 , 92 frames in total
Add frame 91 , 92 frames in total
Computing global features
Traceback (most recent call last):
  File "generate_real_video.py", line 146, in <module>
    framework.compute()
  File "/home/developer/demo/ReReVST-Code/test/framework.py", line 93, in compute
    self.model.compute()
  File "/home/developer/demo/ReReVST-Code/test/style_network_global.py", line 478, in compute
    self.Decoder.compute(torch.cat(self.F_patches, dim=0), self.F_style)
  File "/home/developer/demo/ReReVST-Code/test/style_network_global.py", line 429, in compute
    h = self.slice4.compute(h)
  File "/home/developer/demo/ReReVST-Code/test/style_network_global.py", line 127, in compute
    x_s = self.conv_shortcut(x)
  File "/home/developer/anaconda3/envs/vst/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/developer/anaconda3/envs/vst/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 399, in forward
    return self._conv_forward(input, self.weight, self.bias)
  File "/home/developer/anaconda3/envs/vst/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 395, in _conv_forward
    return F.conv2d(input, weight, bias, self.stride,
RuntimeError: CUDA out of memory. Tried to allocate 5.11 GiB (GPU 0; 23.70 GiB total capacity; 17.94 GiB already allocated; 4.29 GiB free; 18.08 GiB reserved in total by PyTorch)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions