diff --git a/README.md b/README.md index 497a45358..7651c8196 100644 --- a/README.md +++ b/README.md @@ -135,6 +135,19 @@ You can run the tests (after installation) with pytest tests +GPU-specific tests require additional dependencies specified in `continuous_integration/gpuci/environment.yaml`. +These can be added to the development environment by running + +``` +conda env update -n dask-sql -f continuous_integration/gpuci/environment.yaml +``` + +And GPU-specific tests can be run with + +``` +pytest tests -m gpu --rungpu +``` + ## SQL Server `dask-sql` comes with a small test implementation for a SQL server. diff --git a/continuous_integration/gpuci/environment.yaml b/continuous_integration/gpuci/environment.yaml new file mode 100644 index 000000000..efceccb2f --- /dev/null +++ b/continuous_integration/gpuci/environment.yaml @@ -0,0 +1,17 @@ +name: gpuci +channels: + - rapidsai + - rapidsai-nightly + - nvidia +dependencies: + - rust>=1.60.0 + - setuptools-rust>=1.2.0 + - cudatoolkit=11.5 + - cudf=22.08 + - cuml=22.08 + - dask-cudf=22.08 + - dask-cuda=22.08 + - numpy>=1.20.0 + - ucx-proc=*=gpu + - ucx-py=0.27 + - xgboost=*=cuda_* diff --git a/docs/source/installation.rst b/docs/source/installation.rst index e082dc893..8c194fd2c 100644 --- a/docs/source/installation.rst +++ b/docs/source/installation.rst @@ -109,6 +109,19 @@ You can run the tests (after installation) with pytest tests +GPU-specific tests require additional dependencies specified in `continuous_integration/gpuci/environment.yaml`. +These can be added to the development environment by running + +.. code-block:: bash + + conda env update -n dask-sql -f continuous_integration/gpuci/environment.yaml + +And GPU-specific tests can be run with + +.. code-block:: bash + + pytest tests -m gpu --rungpu + This repository uses pre-commit hooks. To install them, call .. code-block:: bash