search

CannyLab / tsne-cuda

GPU Accelerated t-SNE for CUDA with Python bindings
BSD 3-Clause "New" or "Revised" License
1.8k stars 129 forks source link

MultiGPU support #106

Closed ferrine closed 3 years ago

ferrine commented 3 years ago

Hey, I'm using this great repo on really massive data with great success. Many thanks for your work.

Do you have plans for multigpu support so it can handle more data or work even faster?

DavidMChan commented 3 years ago

This library already supports multi-gpu in sharded search when computing the nearest neighbors (I think by default it will use all available GPUs), however multi-gpu t-SNE is non-trivial, and not something that's currently on our roadmap. We'd love contributions, however!