如何安装 dask_cudf？

问题描述

我在终端中使用以下几行来安装 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 notebook 时：

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'

为什么找不到模块？

解决方法

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 Toolkit 版本。您可以在 Getting Started 页面上找到此命令（和其他安装选项）。

您可能还想仔细检查您使用的 Python 是否实际上是来自 Conda 环境的 Python。这通常可以通过 which python 完成。