benfred / implicit

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

Error: cuda extension is built, but disabling gpu support because of '{e}' #558

Closed caijunjieNick closed 2 years ago

benfred commented 2 years ago

Usually this warning is set because you don't have a GPU installed, but you have a valid CUDA toolkit. If you don't have CUDA/cublas etc you shouldn't see this warning at all , and GPU support will just be disabled.

You can filter out this warning by running warnings.filterwarnings("ignore", module="implicit.gpu") in your code