search

youngLBW / HRN

[CVPR2023] A Hierarchical Representation Network for Accurate and Detailed Face Reconstruction from In-The-Wild Images.
https://younglbw.github.io/HRN-homepage/
Apache License 2.0
407 stars 37 forks source link

ninja: build stopped: subcommand failed. #53

Open utsunai opened 7 months ago

utsunai commented 7 months ago

I met this error, Plz help if you know what's going on thank you!

Traceback (most recent call last):
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1509, in _run_ninja_build
    subprocess.run(
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/subprocess.py", line 516, in run
    raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "demo.py", line 78, in <module>
    run_hrn(args)
  File "demo.py", line 26, in run_hrn
    output = reconstructor.predict(img, visualize=True, save_name=save_name, out_dir=out_dir)
  File "/home/tianhy/HRN/models/hrn.py", line 238, in predict
    output = self.predict_base(img)
  File "/home/tianhy/HRN/models/hrn.py", line 165, in predict_base
    output = self.model.predict_results_base()  # run inference
  File "/home/tianhy/HRN/models/facerecon_model.py", line 278, in predict_results_base
    pred_mask, _, pred_face = self.renderer.render_uv_texture(face_vertex, self.facemodel_front.face_buf,
  File "/home/tianhy/HRN/util/nv_diffrast.py", line 173, in render_uv_texture
    self.glctx = dr.RasterizeGLContext(device=device)
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/torch/ops.py", line 150, in __init__
    self.cpp_wrapper = _get_plugin().RasterizeGLStateWrapper(output_db, mode == 'automatic', cuda_device_idx)
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/torch/ops.py", line 83, in _get_plugin
    torch.utils.cpp_extension.load(name=plugin_name, sources=source_paths, extra_cflags=opts, extra_cuda_cflags=opts, extra_ldflags=ldflags, with_cuda=True, verbose=False)
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 963, in load
    return _jit_compile(
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1170, in _jit_compile
    _write_ninja_file_and_build_library(
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1276, in _write_ninja_file_and_build_library
    _run_ninja_build(
  File "/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1529, in _run_ninja_build
    raise RuntimeError(message)
RuntimeError: Error building extension 'nvdiffrast_plugin': [1/5] /home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/rasterize.cu -o rasterize.cuda.o
FAILED: rasterize.cuda.o
/home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/rasterize.cu -o rasterize.cuda.o
/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/rasterize.cu(19): warning #177-D: variable "_ca_temp" was declared but never referenced
      float _ca_temp;
            ^

Remark: The warnings can be suppressed with "-diag-suppress <warning-number>"

ptxas /tmp/tmpxft_0000333a_00000000-6_rasterize.ptx, line 9; fatal   : Unsupported .version 8.3; current version is '7.7'
ptxas fatal   : Ptx assembly aborted due to errors
[2/5] /home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/interpolate.cu -o interpolate.cuda.o
FAILED: interpolate.cuda.o
/home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/interpolate.cu -o interpolate.cuda.o
/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/interpolate.cu(135): warning #177-D: variable "_ca_temp" was declared but never referenced
      float _ca_temp;
            ^

Remark: The warnings can be suppressed with "-diag-suppress <warning-number>"

ptxas /tmp/tmpxft_0000333c_00000000-6_interpolate.ptx, line 9; fatal   : Unsupported .version 8.3; current version is '7.7'
ptxas fatal   : Ptx assembly aborted due to errors
[3/5] /home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/antialias.cu -o antialias.cuda.o
FAILED: antialias.cuda.o
/home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/antialias.cu -o antialias.cuda.o
/home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/antialias.cu(390): warning #177-D: variable "_ca_temp" was declared but never referenced
      float _ca_temp;
            ^

Remark: The warnings can be suppressed with "-diag-suppress <warning-number>"

ptxas /tmp/tmpxft_0000333e_00000000-6_antialias.ptx, line 9; fatal   : Unsupported .version 8.3; current version is '7.7'
ptxas fatal   : Ptx assembly aborted due to errors
[4/5] /home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/texture.cu -o texture.cuda.o
FAILED: texture.cuda.o
/home/tianhy/miniconda3/envs/HRN/bin/nvcc -DTORCH_EXTENSION_NAME=nvdiffrast_plugin -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/TH -isystem /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/torch/include/THC -isystem /home/tianhy/miniconda3/envs/HRN/include -isystem /home/tianhy/miniconda3/envs/HRN/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -DNVDR_TORCH -std=c++14 -c /home/tianhy/miniconda3/envs/HRN/lib/python3.8/site-packages/nvdiffrast/common/texture.cu -o texture.cuda.o
ptxas /tmp/tmpxft_0000333d_00000000-6_texture.ptx, line 9; fatal   : Unsupported .version 8.3; current version is '7.7'
ptxas fatal   : Ptx assembly aborted due to errors
ninja: build stopped: subcommand failed.
youngLBW commented 6 months ago

We haven't had this problem before. Maybe you can try to change all the "dr.RasterizeGLContext" in util/nv_diffrast.py into "dr.RasterizeCudaContext".