如何安装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 install 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环境的Python。这通常可以使用which python来完成。