benfred / implicit

Fast Python Collaborative Filtering for Implicit Feedback Datasets
https://benfred.github.io/implicit/
MIT License
3.57k stars 612 forks source link

Installation process hangs for 0.5.2 #537

Closed chekhovana closed 2 years ago

chekhovana commented 2 years ago

pip installation starts and never ends, its true for installing in colab or locally on macos Screenshot 2022-01-30 at 13 00 32

benfred commented 2 years ago

Can you try running this with pip install implicit -v to display debug output? I'm suspecting that its not hanging, just talking longer than expected. The packages on pip are doing a source install - so by going pip install implicit you need to compile all the code - and then new version includes GPU inference code that can take some time to install.

You can speed this up a bit by only compiling for the GPU architecture that you're on - for instance if you're on colab and have a K80 gpu you can go IMPLICIT_CUDA_ARCH=37 pip install implicit -vv which should finish quite a bit faster

chekhovana commented 2 years ago

Yes, you are right, according to debug output, it just takes too long to compile.

Thank you.

benfred commented 2 years ago

@chekhovana - I've added pre-built binaries to pypi with #546 , this should install in a couple of seconds on google colab now,

chekhovana commented 2 years ago

Thanks a lot, Ben