search

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)
https://arxiv.org/abs/1904.01355
Other
3.27k stars 629 forks source link

Fix AT_CHECK deprecated since torch 1.5 #361

Open guigarfr opened 3 years ago

guigarfr commented 3 years ago

Fixes this issue

AT_CHECK is deprecated since torch 1.5, so i added this code for compatibility.

ghost commented 1 year ago

Hi, thank you very much for your contribution, but after I replaced the changed files with your, the error still exists, can you take a look at my errors? running build running build_py running build_ext /home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/utils/cpp_extension.py:813: UserWarning: The detected CUDA version (10.1) has a minor version mismatch with the version that was used to compile PyTorch (10.2). Most likely this shouldn't be a problem. warnings.warn(CUDA_MISMATCH_WARN.format(cuda_str_version, torch.version.cuda)) /home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/utils/cpp_extension.py:820: UserWarning: There are no g++ version bounds defined for CUDA version 10.1 warnings.warn(f'There are no {compiler_name} version bounds defined for CUDA version {cuda_str_version}') building 'fcos_core._C' extension /home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/cuda/init.py:146: UserWarning: NVIDIA GeForce RTX 3080 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37. If you want to use the NVIDIA GeForce RTX 3080 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name)) Emitting ninja build file /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/build.ninja... Compiling objects... Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N) [1/9] c++ -MMD -MF /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/ROIAlign_cpu.o.d -pthread -B /home/zhou/anaconda3/envs/FCOSenv/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cpu/ROIAlign_cpu.cpp -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/ROIAlign_cpu.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/ROIAlign_cpu.o c++ -MMD -MF /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/ROIAlign_cpu.o.d -pthread -B /home/zhou/anaconda3/envs/FCOSenv/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cpu/ROIAlign_cpu.cpp -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/ROIAlign_cpu.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cpu/ROIAlign_cpu.cpp:2:10: fatal error: cpu/vision.h: No such file or directory 2 | #include "cpu/vision.h" | ^~~~~~ compilation terminated. [2/9] c++ -MMD -MF /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/nms_cpu.o.d -pthread -B /home/zhou/anaconda3/envs/FCOSenv/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cpu/nms_cpu.cpp -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/nms_cpu.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/nms_cpu.o c++ -MMD -MF /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/nms_cpu.o.d -pthread -B /home/zhou/anaconda3/envs/FCOSenv/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -O2 -isystem /home/zhou/anaconda3/envs/FCOSenv/include -fPIC -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cpu/nms_cpu.cpp -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cpu/nms_cpu.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cpu/nms_cpu.cpp:2:10: fatal error: cpu/vision.h: No such file or directory 2 | #include "cpu/vision.h" | ^~~~~~ compilation terminated. [3/9] /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/nms.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/nms.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/nms.o /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/nms.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/nms.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/nms.cu:5:10: fatal error: THC/THC.h: No such file or directory

include <THC/THC.h>

      ^~~~~~~~~~~

compilation terminated. [4/9] /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/deform_conv_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/deform_conv_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/deform_conv_cuda.o /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/deform_conv_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/deform_conv_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/deform_conv_cuda.cu:7:10: fatal error: THC/THC.h: No such file or directory

include <THC/THC.h>

      ^~~~~~~~~~~

compilation terminated. [5/9] /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ROIAlign_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ROIAlign_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ROIAlign_cuda.o /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ROIAlign_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ROIAlign_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ROIAlign_cuda.cu:5:10: fatal error: THC/THC.h: No such file or directory

include <THC/THC.h>

      ^~~~~~~~~~~

compilation terminated. [6/9] /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/SigmoidFocalLoss_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/SigmoidFocalLoss_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/SigmoidFocalLoss_cuda.o /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/SigmoidFocalLoss_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/SigmoidFocalLoss_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/SigmoidFocalLoss_cuda.cu:8:10: fatal error: THC/THC.h: No such file or directory

include <THC/THC.h>

      ^~~~~~~~~~~

compilation terminated. [7/9] /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ml_nms.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ml_nms.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ml_nms.o /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ml_nms.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ml_nms.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ml_nms.cu:5:10: fatal error: THC/THC.h: No such file or directory

include <THC/THC.h>

      ^~~~~~~~~~~

compilation terminated. [8/9] /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/deform_pool_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/deform_pool_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/deform_pool_cuda.o /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/deform_pool_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/deform_pool_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/deform_pool_cuda.cu:11:10: fatal error: THC/THC.h: No such file or directory

include <THC/THC.h>

      ^~~~~~~~~~~

compilation terminated. [9/9] /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ROIPool_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ROIPool_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 FAILED: /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ROIPool_cuda.o /usr/bin/nvcc -DWITH_CUDA -Ifcos_core/csrc -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/torch/csrc/api/include -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/TH -I/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/include/THC -I/home/zhou/anaconda3/envs/FCOSenv/include/python3.10 -c -c /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ROIPool_cuda.cu -o /home/zhou/wyx/FCOS/build/temp.linux-x86_64-cpython-310/fcos_core/csrc/cuda/ROIPool_cuda.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14 /home/zhou/wyx/FCOS/fcos_core/csrc/cuda/ROIPool_cuda.cu:5:10: fatal error: THC/THC.h: No such file or directory

include <THC/THC.h>

      ^~~~~~~~~~~

compilation terminated. ninja: build stopped: subcommand failed. Traceback (most recent call last): File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1808, in _run_ninja_build subprocess.run( File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/subprocess.py", line 524, 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 "/home/zhou/wyx/FCOS/setup.py", line 66, in setup( File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/init.py", line 87, in setup return distutils.core.setup(**attrs) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/core.py", line 185, in setup return run_commands(dist) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/core.py", line 201, in run_commands dist.run_commands() File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 973, in run_commands self.run_command(cmd) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/dist.py", line 1217, in run_command super().run_command(command) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 992, in run_command cmd_obj.run() File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/command/build.py", line 24, in run super().run() File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/command/build.py", line 132, in run self.run_command(cmd_name) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/cmd.py", line 319, in run_command self.distribution.run_command(command) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/dist.py", line 1217, in run_command super().run_command(command) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/dist.py", line 992, in run_command cmd_obj.run() File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/command/build_ext.py", line 79, in run _build_ext.run(self) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/Cython/Distutils/old_build_ext.py", line 186, in run _build_ext.build_ext.run(self) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 346, in run self.build_extensions() File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 765, in build_extensions build_ext.build_extensions(self) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/Cython/Distutils/old_build_ext.py", line 195, in build_extensions _build_ext.build_ext.build_extensions(self) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 466, in build_extensions self._build_extensions_serial() File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 492, in _build_extensions_serial self.build_extension(ext) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/command/build_ext.py", line 202, in build_extension _build_ext.build_extension(self, ext) File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/setuptools/_distutils/command/build_ext.py", line 547, in build_extension objects = self.compiler.compile( File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 586, in unix_wrap_ninja_compile _write_ninja_file_and_compile_objects( File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1487, in _write_ninja_file_and_compile_objects _run_ninja_build( File "/home/zhou/anaconda3/envs/FCOSenv/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1824, in _run_ninja_build raise RuntimeError(message) from e RuntimeError: Error compiling objects for extension

Thank you very much!