about the "knn_cuda" #33
Description
(M3DM) wangwei-uestc@user-SYS-4029GP-TRT:/datadisk/wangwei/M3DM$ python main.py --method_name DINO+Point_MAE --memory_bank multiple --rgb_backbone_name vit_base_patch8_224_dino --xyz_backbone_name Point_MAE --save_feature
Traceback (most recent call last):
File "/home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1666, in _run_ninja_build
subprocess.run(
File "/home/wangwei-uestc/anaconda3/envs/M3DM/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.
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "main.py", line 2, in
from m3dm_runner import M3DM
File "/datadisk/wangwei/M3DM/m3dm_runner.py", line 5, in
from feature_extractors import multiple_features
File "/datadisk/wangwei/M3DM/feature_extractors/multiple_features.py", line 2, in
from feature_extractors.features import Features
File "/datadisk/wangwei/M3DM/feature_extractors/features.py", line 18, in
from knn_cuda import KNN
File "/home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/knn_cuda/init.py", line 38, in
knn = load_cpp_ext("knn")
File "/home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/knn_cuda/init.py", line 26, in load_cpp_ext
ext = load(
File "/home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1080, in load
return jit_compile(
File "/home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1293, in jit_compile
write_ninja_file_and_build_library(
File "/home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1405, in write_ninja_file_and_build_library
run_ninja_build(
File "/home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1682, in run_ninja_build
raise RuntimeError(message) from e
RuntimeError: Error building extension 'knn': [1/2] /usr/bin/nvcc -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include/TH -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include/THC -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS -D__CUDA_NO_BFLOAT16_CONVERSIONS -D__CUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 --compiler-options '-fPIC' -DCUDA_HAS_FP16=1 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ -std=c++14 -c /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cu -o knn.cuda.o
FAILED: knn.cuda.o
/usr/bin/nvcc -DTORCH_EXTENSION_NAME=knn -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include/TH -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/torch/include/THC -isystem /home/wangwei-uestc/anaconda3/envs/M3DM/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 --compiler-options '-fPIC' -DCUDA_HAS_FP16=1 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ -std=c++14 -c /home/wangwei-uestc/anaconda3/envs/M3DM/lib/python3.8/site-packages/knn_cuda/csrc/cuda/knn.cu -o knn.cuda.o
nvcc fatal : Unsupported gpu architecture 'compute_86'
ninja: build stopped: subcommand failed.