Upgrade cuML and cuDF - Githubissues

Kaggle / docker-python

Kaggle Python docker image

Apache License 2.0

2.47k stars 955 forks source link

Upgrade cuML and cuDF #1395

Closed calderjo closed 6 months ago

calderjo commented 6 months ago

upgrade RAPIDS to 24.4.x (cuML and cuDF).

looks like installing via mamba was not working, installed using conda with mamba package solver (which is faster than plan conda).
Added test for cuML and cuDF, these packages were reported as "broken" for last the 6 months.
Skip certain test if running on a p100 gpu

Going forward, rapids will not be compatible with p100 GPUs, that's a problem for another day https://chat.kaggle.net/kaggle/pl/85tczsc4w3nhijkd1ftryxr7yo b/296444923 b/341938540

djherbis commented 6 months ago

Great work, thanks :)