search

unlimblue / KNN_CUDA

pytorch knn [cuda version]
292 stars 37 forks source link

Import error #2

Closed JeremyFisher closed 5 years ago

JeremyFisher commented 5 years ago

Hey,

when I try to from knn_cuda import KNN I get the following error:

ImportError: (...)/KNN_CUDA/knn_cuda/csrc/_ext/knn/knn.so: undefined symbol: _ZN3c105ErrorC1ENS_14SourceLocationERKSs

Any ideas how to fix that?

unlimblue commented 5 years ago

which torch version did you use?

JeremyFisher commented 5 years ago

pytorch 1.0.1 cuda90py36h8b0c50b_0

unlimblue commented 5 years ago

I can't reproduce it , but you can try upgrade your torch to 1.1.0 and then

pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.1/KNN_CUDA-0.1-py3-none-any.whl
JeremyFisher commented 5 years ago

Thanks! It worked!