TypeError: knnquery_cuda()

[0belle]

File "/home/RepSurf/classification/modules/pointnet2_utils.py", line 110, in query_knn_point return knnquery(k, xyz, new_xyz) File "/home/RepSurf/classification/modules/pointops/functions/pointops.py", line 315, in forward pointops_cuda.knnquery_cuda(b, n, m, nsample, xyz, new_xyz, idx, dist2) TypeError: knnquery_cuda(): incompatible function arguments. The following argument types are supported: 1. (arg0: int, arg1: int, arg2: at::Tensor, arg3: at::Tensor, arg4: at::Tensor, arg5: at::Tensor, arg6: at::Tensor, arg7: at::Tensor) -> None

[XuHan-CN]

File "/home/RepSurf/classification/modules/pointnet2_utils.py", line 110, in query_knn_point return knnquery(k, xyz, new_xyz) File "/home/RepSurf/classification/modules/pointops/functions/pointops.py", line 315, in forward pointops_cuda.knnquery_cuda(b, n, m, nsample, xyz, new_xyz, idx, dist2) TypeError: knnquery_cuda(): incompatible function arguments. The following argument types are supported:

1. (arg0: int, arg1: int, arg2: at::Tensor, arg3: at::Tensor, arg4: at::Tensor, arg5: at::Tensor, arg6: at::Tensor, arg7: at::Tensor) -> None

I have also encountered this problem. May I ask how you resolved it

[ZHIBINJIAN]

Could you please provide a solution to this issue? I have been stuck on it for quite some time. If there is a solution, please do inform me. Thank you."

[XuHan-CN]

You have to create a new environment and rebuild the pointops. There are something different between the classification task and segmentation task.