search

src-d / kmcuda

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
Other
783 stars 144 forks source link

Can't install on Google colab #98

Closed j-bac closed 4 years ago

j-bac commented 4 years ago

I would like to try the package on Colab but I didn't manage to install it with: !CUDA_ARCH=75 pip install git+https://github.com/src-d/kmcuda.git#subdirectory=src or !git clone https://github.com/src-d/kmcuda !cd kmcuda/src && python setup.py install

error log: 'CMake Error at /usr/local/lib/python2.7/dist-packages/cmake/data/share/cmake-3.12/Modules/FindPackageHandleStandardArgs.cmake:137 (message):', ' Could NOT find CUDA (missing: CUDA_INCLUDE_DIRS CUDA_CUDART_LIBRARY) (found', ' version "10.0")'

vmarkovtsev commented 4 years ago

Hi @j-bac please refer to Building. There is an env var that you should define.

Closing this...