bash make_all.sh

[cherryolg]

When I perform"sudo bash make_all.sh",
Compiling rod_align kernels by nvcc... In file included from rod_align_kernel.cu:2:0: rod_align_kernel.h:4:21: fatal error: THC/THC.h: No such file or directory #include <THC/THC.h> ^ compilation terminated.
FAILED: /home/zm/wd/GAIC/untils/rod_align/build/temp.linux-x86_64-3.6/src/rod_align_kernel.o
Could you help me see what's wrong？

[aquastripe]

The THC.h library is deprecated by PyTorch. Just replace all

#include <THC/THC.h>

with

#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAEvent.h>

Like this: https://github.com/aquastripe/gaic-lld533/blob/master/roi_align/src/roi_align_kernel.h