git clone https://github.com/chrischoy/pytorch_knn_cuda
cd pytorch_knn_cuda
make
python __init__.py
andrei-pokrovsky / pytorch_knn_cuda Goto Github PK
View Code? Open in Web Editor NEWThis project forked from chrischoy/pytorch_knn_cuda
K-Nearest Neighbor in Pytorch