RuntimeError: Not compiled with GPU support
Opened this issue · 0 comments
Darrenxc commented
After i did these modification #169.
Change the lib/knn/init.py and l ib/loss.py showing below
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
torch version
ubuntu 16.04
Can someone helps? Thank you