vincenty-cuda-nns 0.1.2

Nearest neighbor search on Earth's surface with a GPU

Vincenty nearest neighbor search using CUDA

Nearest neighbor search algorithm on Earth's surface that runs on a GPU and uses Vincenty's formula

Requirements

• CUDA-enabled GPU with compute capability 2.0 or above with an up-to-data Nvidia driver.
• CUDA toolkit

Installation

pip install vincenty-cuda-nns

Usage example

import geopandas as gpd
from vincenty_cuda_nns import CudaTree

df = gpd.read_file('points.geojson')

# data is array of points like [longitude, latitude]
data = np.stack(df['geometry']).astype(np.float32)

# build tree for the data
cuda_tree = CudaTree(data, leaf_size=4)

# query over the tree for nearest neighbor (including itself)
distances, indices = cuda_tree.query(n_neighbors=2)