Pytorch KNN CUDA Install git clone https://github.com/shaibagon/pytorch_knn_cuda.git cd pytorch_knn_cuda make python __init__.py Notes This repository works with pytorch v0.4.1 Works with batched tensors