diff --git a/dask_sql/environment.yml b/dask_sql/environment.yml index 0ddf750..1ca58ce 100644 --- a/dask_sql/environment.yml +++ b/dask_sql/environment.yml @@ -14,3 +14,6 @@ dependencies: - ucx-proc=*=gpu - ucx-py=UCX_PY_VER - xgboost=*=cuda_* +# TODO: remove once cuML is using consolidated RAFT packages +- libfaiss>=1.7.1 +- faiss-proc=*=cuda