j96w / DenseFusion

"DenseFusion: 6D Object Pose Estimation by Iterative Dense Fusion" code repository
https://sites.google.com/view/densefusion
MIT License
1.08k stars 301 forks source link

RuntimeError: Not compiled with GPU support #221

Open Darrenxc opened 1 year ago

Darrenxc commented 1 year ago

After i did these modification #169. Change the lib/knn/init.py and l ib/loss.py showing below image

if not refine:
    if idx[0].item() in sym_list:
        target = target[0].transpose(1, 0).contiguous().view(3, -1)
        pred = pred.permute(2, 0, 1).contiguous().view(3, -1)
        # inds = knn(target.unsqueeze(0), pred.unsqueeze(0))
        inds = KNearestNeighbor.apply(target.unsqueeze(0), pred.unsqueeze(0))
        target = torch.index_select(target, 1, inds.view(-1).detach() - 1)
        target = target.view(3, bs * num_p, num_point_mesh).permute(1, 2, 0).contiguous()
        pred = pred.view(3, bs * num_p, num_point_mesh).permute(1, 2, 0).contiguous()

my system image torch version image ubuntu 16.04

i got this error: image

Can someone helps? Thank you