Skip to content

Commit

Permalink
Add python 3.11 GPU CI job
Browse files Browse the repository at this point in the history
  • Loading branch information
charlesbluca committed May 24, 2024
1 parent 7452d14 commit 33d69ee
Show file tree
Hide file tree
Showing 2 changed files with 54 additions and 0 deletions.
1 change: 1 addition & 0 deletions continuous_integration/gpuci/axis.yaml
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
PYTHON_VER:
- "3.9"
- "3.10"
- "3.11"

CUDA_VER:
- "11.8.0"
Expand Down
53 changes: 53 additions & 0 deletions continuous_integration/gpuci/environment-3.11.yaml
Original file line number Diff line number Diff line change
@@ -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*

0 comments on commit 33d69ee

Please sign in to comment.