如何安装 dask_cudf？

Question

我在终端中使用以下几行来安装 rapids 然后 dask cudf:

conda create -n rapids-core-0.14 -c rapidsai -c nvidia -c conda-forge \
     -c defaults rapids=0.14 python=3.7 cudatoolkit=10.1
conda activate rapids-core-0.14
conda install -c rapidsai dask-cudf

但是当我将 dask_cudf 导入 python 笔记本时：

import dask_cudf

我收到以下错误：

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-1-827f6c5d8016> in <module>
      9 from dask.distributed import Client
     10 from dask.utils import parse_bytes
---> 11 import dask_cudf
     12 np.set_printoptions(formatter={'float': lambda x: "{0:0.4f}".format(x)})
     13 pd.set_option('display.float_format', lambda x: '%.5f' % x)

ModuleNotFoundError: No module named 'dask_cudf'

为什么找不到模块？

Answer 1

Conda 安装 RAPIDS 的推荐安装命令如下：

conda create -n rapids-0.17 -c rapidsai -c nvidia -c conda-forge \
    -c defaults rapids-blazing=0.17 python=3.7 cudatoolkit=10.1

请注意，您可能需要根据您的系统更改 CUDA 工具包版本。您可以在 Getting Started 页面上找到此命令（和其他安装选项）。

您可能还需要仔细检查您正在使用的 Python 是否确实来自 Conda 环境。这通常可以用 which python.

来完成

如何安装 dask_cudf？

How do I install dask_cudf?

python

installation

dask

rapids

cudf