get error when run demo.py dcn_v2.so:undefined symbol:_cudaRegisterFatBinaryEnd

xingyizhou / pytorch-pose-hg-3d

PyTorch implementation for 3D human pose estimation

GNU General Public License v3.0

615 stars 141 forks source link

Open jianghe9 opened 5 years ago

jianghe9 commented 5 years ago

cuda:9.2 pytorch:0.4.1 python demo.py ctdet --demo /path/to/image/or/folder/or/video --load_model ../models/ctdet_coco_dla_2x.pth

dcn_v2.so:undefined symbol:_cudaRegisterFatBinaryEnd

Xulei-Zhai commented 3 years ago

@jianghe9 So, have you solved this problem?

Xulei-Zhai commented 3 years ago

I have sloved this problem with DCNv2 for pytorch 1.x