安装

Conda

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

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

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

一旦确定了合适的CUDA Toolkit版本，可以使用conda与Dask-CUDA一起安装。 要安装最新版本的Dask-CUDA以及CUDA Toolkit 12.5：

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

Pip

在Conda环境之外工作时，可以从NVIDIA的网站下载并安装CUDA Toolkit；这个包也包含了所需的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