j96w/DenseFusion

RuntimeError: Not compiled with GPU support

Opened this issue · 0 comments

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