rapidsai / dask-build-environment

Build environments for various dask related projects on gpuCI
4 stars 11 forks source link

Always use RAPIDS-published xgboost in dask-sql images #63

Closed charlesbluca closed 1 year ago

charlesbluca commented 1 year ago

Recently, the build string of RAPIDS-published XGBoost nightlies changed slightly, which caused the conda environment solve to start failing on builds of the dask-sql images.

This PR modifies the XGBoost specification to fix this failure, and pins us explicitly to the RAPIDS-published packages, as doing a specification like xgboost=*=cuda* would allow us to fallback on the conda-forge packages, which I don't think we want here (cc @VibhuJawa if you have any thoughts here).

Also removing the libfaiss/faiss-proc installs as those should no longer be necessary

VibhuJawa commented 1 year ago

This looks good to me, this will allow users to use RMM with XGboost+Dask-sql which is a good side effect.