diff --git a/continuous_integration/gpuci/axis.yaml b/continuous_integration/gpuci/axis.yaml index 267eb5a24..f7dff8116 100644 --- a/continuous_integration/gpuci/axis.yaml +++ b/continuous_integration/gpuci/axis.yaml @@ -1,6 +1,7 @@ PYTHON_VER: - "3.9" - "3.10" +- "3.11" CUDA_VER: - "11.8.0" diff --git a/continuous_integration/gpuci/environment-3.11.yaml b/continuous_integration/gpuci/environment-3.11.yaml new file mode 100644 index 000000000..ad1fa43d6 --- /dev/null +++ b/continuous_integration/gpuci/environment-3.11.yaml @@ -0,0 +1,53 @@ +name: dask-sql +channels: +- rapidsai +- rapidsai-nightly +- dask/label/dev +- conda-forge +- nvidia +- nodefaults +dependencies: +- c-compiler +- zlib +- dask>=2024.4.1 +- dask-expr>=1.0.11 +- fastapi>=0.92.0 +- fugue>=0.7.3 +- httpx>=0.24.1 +- intake>=0.6.0 +- jsonschema +- lightgbm +- maturin>=1.3,<1.4 +- mock +- numpy>=1.22.4 +- pandas>=2 +- pre-commit +- prompt_toolkit>=3.0.8 +- psycopg2 +- pyarrow>=14.0.1 +- pygments>=2.7.1 +- pyhive +- pytest-cov +- pytest-rerunfailures +- pytest-xdist +- pytest +- python=3.11 +- py-xgboost>=2.0.3 +- scikit-learn>=1.0.0 +- sphinx +- sqlalchemy +- tpot>=0.12.0 +# FIXME: https://github.com/fugue-project/fugue/issues/526 +- triad<0.9.2 +- tzlocal>=2.1 +- uvicorn>=0.14 +# GPU-specific requirements +- cudatoolkit=11.8 +- cudf=24.06 +- cuml=24.06 +- dask-cudf=24.06 +- dask-cuda=24.06 +- ucx-proc=*=gpu +- ucx-py=0.38 +- xgboost=*=rapidsai_py* +- libxgboost=*=rapidsai_h*