Pytorch1.5编译DCNv2时报错:error: identifier “THCState_getCurrentStream“ is undefined

应该是pytorch版本的问题,有网友将pytorch切换到1.4便解决了。现在官方给出了解决方法,可见DCNv2-git。
简单来说就是将src/cuda/dcn_v2_cuda.cu 文件中的THCState_getCurrentStream(state)都替换为c10::cuda::getCurrentCUDAStream()。

不同pytorch版本的DCNv2

你可能感兴趣的:(算法,配置)