如何安装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完成