lucasjinreal / DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)

BSD 3-Clause "New" or "Revised" License

609 stars 125 forks source link

what's ATen? #54

Closed shaqing closed 2 years ago

shaqing commented 2 years ago

i follow solution in #54, but i met this problem: home/sti/kevin/work/smoke/SMOKE/smoke/csrc/cpu/dcn_v2_psroi_pooling_cpu.cpp:16:10: fatal error: ATen/ceil_div.h: No such file or directory

include <ATen/ceil_div.h>

      ^~~~~~~~~~~~~~~~~

compilation terminated.

lucasjinreal commented 2 years ago

please read readme first.

juanmed commented 2 years ago

Hi, I found the same problem as @shaqing on the master branch using pytorch1.8.1+cuda111 Ubuntu18.04. When I try pytorch1.6 branch, I get the following error:

 error: ‘THFloatBlas_gemm’ was not declared in this scope
         THFloatBlas_gemm('n', 't', n, m, k, 1.0f,

I understand you tested on Ubuntu20 and recommend using pytorch1.11. I cannot change my pytorch/cuda so I was wondering if you have any comments?

lucasjinreal commented 2 years ago

Am not going support old pytorch anylonger, but you can switch to older branch for luck.

juanmed commented 2 years ago

I understand. Thank you very much!