CenterPose icon indicating copy to clipboard operation
CenterPose copied to clipboard

Published 20 hours ago • verified publisher icon NVlabs

Run with CPU

Open nuaayxy opened this issue 2 years ago • 4 comments

is there a way to run the demo with Cpu only? I am having so many errors getting the DCNv2 working with cuda......

nuaayxy avatar May 28 '22 03:05 nuaayxy

Sorry. I have not tried CPU-only before. Maybe it has some trouble with DCNv2. You can try our provided weights with resnet instead.

Uio96 avatar Jun 14 '22 23:06 Uio96

Tried resnet still not working

File "/home/usr/CenterPose/src/lib/models/networks/DCNv2/dcn_v2.py", line 128, in forward self.deformable_groups) File "/home/usr/CenterPose/src/lib/models/networks/DCNv2/dcn_v2.py", line 31, in forward ctx.deformable_groups) RuntimeError: Not compiled with GPU support (dcn_v2_forward at /home/usr/CenterPose/src/lib/models/networks/DCNv2/src/dcn_v2.h:35)

nuaayxy avatar Jun 23 '22 07:06 nuaayxy

Tried resnet still not working

File "/home/usr/CenterPose/src/lib/models/networks/DCNv2/dcn_v2.py", line 128, in forward self.deformable_groups) File "/home/usr/CenterPose/src/lib/models/networks/DCNv2/dcn_v2.py", line 31, in forward ctx.deformable_groups) RuntimeError: Not compiled with GPU support (dcn_v2_forward at /home/usr/CenterPose/src/lib/models/networks/DCNv2/src/dcn_v2.h:35)

You probably have to comment the codes related to DCNv2 (e.g., import stuff), then have another try.

Uio96 avatar Jun 23 '22 14:06 Uio96

code is tightly coupled with DCNv2 everywhere, no way to comment it out and have working sample

nuaayxy avatar Jun 28 '22 22:06 nuaayxy