search

ParthaEth / GIF

GIF is a photorealistic generative face model with explicit 3D geometric and photometric control.
https://gif.is.tue.mpg.de/
MIT License
405 stars 63 forks source link

RuntimeError: Not compiled with GPU support #30

Closed arnobanu closed 10 months ago

arnobanu commented 10 months ago

Dear authors, thank you so much for sharing your work. I wanted to reproduce the results by training the model from scratch. I have installed the packages in the requirements.txt in a python 3.8.18 environment but I am getting error like:


Exception has occurred: RuntimeError       (note: full exception trace is shown but execution is paused at: _run_module_as_main)
Not compiled with GPU support
  File "/data/mdwkhan/3d_gen_codes/pytorch3d/pytorch3d/renderer/mesh/rasterize_meshes.py", line 189, in forward
    pix_to_face, zbuf, barycentric_coords, dists = _C.rasterize_meshes(
  File "/data/mdwkhan/3d_gen_codes/pytorch3d/pytorch3d/renderer/mesh/rasterize_meshes.py", line 136, in rasterize_meshes
    return _RasterizeFaceVerts.apply(
  File "/data/mdwkhan/3d_gen_codes/GIF/my_utils/photometric_optimization/renderer.py", line 59, in forward
    pix_to_face, zbuf, bary_coords, dists = rasterize_meshes(
  File "/data/mdwkhan/anaconda3/envs/gif2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/data/mdwkhan/3d_gen_codes/GIF/my_utils/photometric_optimization/renderer.py", line 152, in forward
    rendering = self.rasterizer(transformed_vertices, self.faces.expand(batch_size, -1, -1), attributes)
  File "/data/mdwkhan/anaconda3/envs/gif2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/data/mdwkhan/3d_gen_codes/GIF/my_utils/photometric_optimization/gif_helper.py", line 37, in render_tex_and_normal
    rendering_results = self.render(verts, trans_verts, albedos, lights=lightcode)
  File "/data/mdwkhan/3d_gen_codes/GIF/my_utils/visualize_flame_overlay.py", line 24, in get_rendered_mesh
    self.rendering_helper.render_tex_and_normal(shapecode=shape, expcode=expression,
  File "/data/mdwkhan/3d_gen_codes/GIF/loss_functions/losses.py", line 210, in get_image_and_textures
    self.flame_visualizer.get_rendered_mesh(flame_params=(shape, exp, pose, light_code, texture_code),
  File "/data/mdwkhan/3d_gen_codes/GIF/loss_functions/losses.py", line 163, in tex_sp_intrp_loss
    textures, tx_masks, _ = self.get_image_and_textures(alpha, flame_batch, generator, max_ids, normal_maps_as_cond,
  File "/data/mdwkhan/3d_gen_codes/GIF/train.py", line 229, in train
    interp_loss = interp_tex_loss.tex_sp_intrp_loss(
  File "/data/mdwkhan/3d_gen_codes/GIF/train.py", line 402, in <module>
    train(args, dataset, generator, discriminator_flm, fid_computer, flame_param_est, used_sampless,
  File "/data/mdwkhan/anaconda3/envs/gif2/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/data/mdwkhan/anaconda3/envs/gif2/lib/python3.8/runpy.py", line 194, in _run_module_as_main (Current frame)
    return _run_code(code, main_globals, None,
RuntimeError: Not compiled with GPU support
arnobanu commented 10 months ago

I solved the problem using the main repository of pytorch3d (installation).