安装

Dask-CUDA 可以使用 conda、pip 或从源码进行安装。

Conda

要在系统上使用 Dask-CUDA，您需要

• 您 GPU 的 NVIDIA 驱动程序；有关安装说明，请参阅 NVIDIA 驱动程序安装快速入门指南

• 与已安装驱动程序版本兼容的 NVIDIA CUDA 工具包版本；有关 CUDA 工具包驱动程序要求的概述，请参阅 CUDA 兼容性 – 二进制兼容性 表 1

确定合适的 CUDA 工具包版本后，可以使用 conda 与 Dask-CUDA 一起安装。要安装最新版本的 Dask-CUDA 以及 CUDA 工具包，请执行以下操作

conda install -c rapidsai -c conda-forge -c nvidia dask-cuda cuda-version=12.8

Pip

在 Conda 环境外部工作时，可以从 NVIDIA 网站下载并安装 CUDA 工具包；此软件包也包含所需的 NVIDIA 驱动程序。要安装最新版本的 Dask-CUDA，请执行以下操作

python -m pip install dask-cuda

源码

要从源码安装 Dask-CUDA，必须从 GitHub 克隆源码仓库

git clone https://github.com/rapidsai/dask-cuda.git
cd dask-cuda
python -m pip install .

其他 RAPIDS 库

Dask-CUDA 是 RAPIDS GPU 加速数据科学开源软件库套件的一部分，可与其良好协同工作。有关如何安装这些库的说明，请参阅 RAPIDS – 入门。请记住，这些库将需要

• 至少一个符合 CUDA 标准的 GPU

• 系统已安装 CUDA