Home

Awesome

Pytorch KNN CUDA

git clone https://github.com/chrischoy/pytorch_knn_cuda
cd pytorch_knn_cuda
make
python __init__.py