search

benfred / implicit

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

gpu issue on colab (implicit==0.7.2 / cuda ==12.1) #712

Closed nnfuzzy closed 8 months ago

nnfuzzy commented 8 months ago

Hi,

i don't know exactly why (maybe a colab bump in cuda version?) , because it worked in the past , but I can't use the GPU in colab anymore. Installed wih pip the latest version (0.7.2) .

I'm getting a: ValueError: No CUDA extension has been built, can't train on GPU. But cuda ist available: Is CUDA available? True / CUDA version: 12.1

Thanks in advance, Christian

nnfuzzy commented 8 months ago

It works if you do:

!pip install -q condacolab
import condacolab
condacolab.install()

!conda install -c conda-forge implicit implicit-proc=*=gpu