需要编译 DCNv2. 但是遇到如下错误: DCNv2/src/cuda/dcn_v2_cuda.cu(107): error: identifier "THCState_getCurrentStream" is undefined 简单来说就是将src/cuda/dcn_v2_cuda.cu 文件中的THCState_getCurrentStream(state)都替换为c10::cuda::get
Processing dependenciesforDCNv2==0.1 Finished processing dependenciesforDCNv2==0.1 原因分析:通过docker run进入到docker里面编译时,已通过--gpus选项为docker指定了GPU,所以可以使用GPU并编译成功。但在docker build执行Dockerfile时并未为docker指定GPU,所以依然无法使用GPU。 终极方案:不在docker build时通过Dockerfile...