skhu101 / GauHuman

Code for our CVPR'2024 paper "GauHuman: Articulated Gaussian Splatting from Monocular Human Videos"
Other
322 stars 27 forks source link

Why the code can be run directly in vscode, but a RuntimeError occurs when debugging #19

Closed sycamore27 closed 6 months ago

sycamore27 commented 8 months ago

Why the code can be run directly in vscode, but a RuntimeError occurs when debugging:

发生异常: RuntimeError Error building extension 'knn': [1/3] /usr/local/cuda-11.8:/usr/local/cuda-11.8/bin/nvcc -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_89,code=compute_89 -gencode=arch=compute_89,code=sm_89 --compiler-options '-fPIC' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -std=c++17 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cu -o knn.cuda.o FAILED: knn.cuda.o /usr/local/cuda-11.8:/usr/local/cuda-11.8/bin/nvcc -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_89,code=compute_89 -gencode=arch=compute_89,code=sm_89 --compiler-options '-fPIC' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -std=c++17 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cu -o knn.cuda.o /bin/sh: 1: /usr/local/cuda-11.8:/usr/local/cuda-11.8/bin/nvcc: not found [2/3] c++ -MMD -MF knn.o.d -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -std=c++17 -O2 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cpp -o knn.o FAILED: knn.o c++ -MMD -MF knn.o.d -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -std=c++17 -O2 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cpp -o knn.o In file included from /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cpp:3: /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/ATen/cuda/CUDAContext.h:5:10: fatal error: cuda_runtime_api.h: No such file or directory 5 | #include | ^~~~~~~~ compilation terminated. ninja: build stopped: subcommand failed. subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

The above exception was the direct cause of the following exception:

File "/home/qinzhen/GauHuman/scene/gaussian_model.py", line 23, in from knn_cuda import KNN File "/home/qinzhen/GauHuman/scene/dataset_readers.py", line 28, in from scene.gaussian_model import BasicPointCloud File "/home/qinzhen/GauHuman/scene/init.py", line 16, in from scene.dataset_readers import sceneLoadTypeCallbacks File "/home/qinzhen/GauHuman/gaussian_renderer/init.py", line 15, in from scene.gaussian_model import GaussianModel File "/home/qinzhen/GauHuman/train.py", line 16, in from gaussian_renderer import render, network_gui RuntimeError: Error building extension 'knn': [1/3] /usr/local/cuda-11.8:/usr/local/cuda-11.8/bin/nvcc -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_89,code=compute_89 -gencode=arch=compute_89,code=sm_89 --compiler-options '-fPIC' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -std=c++17 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cu -o knn.cuda.o FAILED: knn.cuda.o /usr/local/cuda-11.8:/usr/local/cuda-11.8/bin/nvcc -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_89,code=compute_89 -gencode=arch=compute_89,code=sm_89 --compiler-options '-fPIC' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -std=c++17 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cu -o knn.cuda.o /bin/sh: 1: /usr/local/cuda-11.8:/usr/local/cuda-11.8/bin/nvcc: not found [2/3] c++ -MMD -MF knn.o.d -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -std=c++17 -O2 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cpp -o knn.o FAILED: knn.o c++ -MMD -MF knn.o.d -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/TH -isystem /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda-11.8:/usr/local/cuda-11.8/include -isystem /home/qinzhen/anaconda3/envs/gauhuman/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -std=c++17 -O2 -c /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cpp -o knn.o In file included from /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cpp:3: /home/qinzhen/anaconda3/envs/gauhuman/lib/python3.8/site-packages/torch/include/ATen/cuda/CUDAContext.h:5:10: fatal error: cuda_runtime_api.h: No such file or directory 5 | #include | ^~~~~~~~ compilation terminated. ninja: build stopped: subcommand failed.

skhu101 commented 8 months ago

Hi, maybe you can try installing pytorch3d and using the KNN in pytorch3d.