Added 1gpu to kmeans.