Compile problem about chamferdistancepytorch

[jih189]

even i changed the pytorch back to 1.4.0, it still does not work

[ThibaultGROUEIX]

Hi @jih189
I don't what might cause this. Please mention it if you find a solution, i'll test the code on my end,
Cheers
Thibault

[Skyfallk]

Get the same error on Pytorch 1.6.0 and 1.5.1 +cuda10.1. Change video driver, reinstall torch, nothing works(

[jih189]

I think I found the solution. Do not run python setup.py install. If you did it, you can pip uninstall it. Just ensure "pip list" does not show it. Then use this package in your project.

[Skyfallk]

I think I found the solution. Do not run python setup.py install. If you did it, you can pip uninstall it. Just ensure "pip list" does not show it. Then use this package in your project.

This really works! Just coometn some code in dist_chamfer_3D.py

chamfer_found = importlib.find_loader("chamfer_3D") is not None
if not chamfer_found:
Cool trick from https://github.com/chrdiller
print("Jitting Chamfer 3D")

from torch.utils.cpp_extension import load
chamfer_3D = load(name="chamfer_3D",
      sources=[
          "/".join(os.path.abspath(__file__).split('/')[:-1] + ["chamfer_cuda.cpp"]),
          "/".join(os.path.abspath(__file__).split('/')[:-1] + ["chamfer3D.cu"]),
          ])
print("Loaded JIT 3D CUDA chamfer distance")

else:
import chamfer_3D
print("Loaded compiled 3D CUDA chamfer distance")

and all works.