maybe matrixto large, I cannot get a result

src-d / kmcuda

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA

Other

784 stars 144 forks source link

maybe matrixto large, I cannot get a result #73

Open littlejiajia opened 5 years ago

littlejiajia commented 5 years ago

I want to know the largest dataset the kmeans algorithm can support. I have a matrix with the size of (1396, 43277), and my GPU is NVIDIA Geforce RTX 2070 with a total memory of 8GB, I run the kmeans_cuda and it runs for a long time and never stops. Is there anyone can tell me the reason?

vmarkovtsev commented 5 years ago

@littlejiajia Try setting the log level to either 1 or 2. The details how to set it are included into the README.