unlimblue / KNN_CUDA

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

Can you update you code to fit the CUDA version 10.1.105? #3

Closed skJack closed 5 years ago

skJack commented 5 years ago

Thanks your code first, it's very useful. But when i update my cuda version to 10.1.105, your code can not import knn_cuda, it shows “ImportError: libcudart.so.10.1 cannot open shared object file:no such file or directory”.I would be very thankful if you update your code or tell me how to solve this bug.