Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cuda out of memory when running main_texture_transfer.py #6

Open
Sibylle-Van-Hove opened this issue Apr 26, 2022 · 0 comments
Open

Comments

@Sibylle-Van-Hove
Copy link

Sibylle-Van-Hove commented Apr 26, 2022

After successfully running the Face Reconstruction & deformation I'm encountering the following error when running the Texture Stylization step (transfer/main_texture_transfer.py) on the provided test data (00016):

Traceback (most recent call last):
  File "./transfer/main_texture_transfer.py", line 199, in <module>
    run(dir_data_sub)
  File "./transfer/main_texture_transfer.py", line 127, in run
    images = renderer_model(azi=azi[aov_id], ele=ele[aov_id])
  File "/home/gpu/anaconda3/envs/3d-portrait-stylization/lib/python3.7/site-packages/torch/nn/modules/module.py", line 547, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/gpu/avatarTech/3D-Portrait-Stylization/texture_style_transfer/transfer/model_ortho_RE.py", line 61, in forward
    elevation=self.elevation, azimuth=self.azimuth, show_forehead=True)
  File "/home/gpu/avatarTech/3D-Portrait-Stylization/texture_style_transfer/neural_renderer/renderer.py", line 185, in render_rgb
    textures = torch.cat((textures, textures.permute((0, 1, 4, 3, 2, 5))), dim=1)
RuntimeError: CUDA out of memory. Tried to allocate 1.36 GiB (GPU 0; 15.90 GiB total capacity; 14.49 GiB already allocated; 869.44 MiB free; 64.90 MiB cached)

What are the GPU requirements to run transfer/main_texture_transfer.py?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant