Skip to content
27 changes: 27 additions & 0 deletions docs/pages/installation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,33 @@ Install the package from the ``conda-forge`` channel:

conda install dask-sql -c conda-forge

Experimental GPU support
^^^^^^^^^^^^^^^^^^^^^^^^

- GPU support is currently tied to the `RAPIDS <https://rapids.ai/>`_ libraries.
- It is in development and generally requires the latest cuDF nightlies.
- It is experimental, so users should expect some bugs or undefined behavior.

Create a new conda environment or use an existing one to install RAPIDS with the chosen methods and packages:
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think we should maybe add a short blurb about GPU support here, bringing up the following to contextualize the env creation line below:

  • GPU support right now is specifically tied to the RAPIDS libraries
  • It is in development, and generally requires the latest nightlies of cuDF
  • It is experimental, and users should still expect some level of bugs / undefined behavior

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @charlesbluca, I'll see to add them!


.. code-block:: bash

conda create --name rapids-env -c rapidsai-nightly -c nvidia -c conda-forge -y cudatoolkit=11.2 cudf=21.12 ucx-py ucx-proc=*=gpu
conda activate rapids-env

Install the stable package from the ``conda-forge`` channel:

.. code-block:: bash

conda install -c conda-forge dask-sql

Or the latest nightly from the ``dask`` channel (currently only available for Linux-based operating systems):

.. code-block:: bash

conda install -c dask/label/dev dask-sql


With ``pip``
------------

Expand Down