0

我在终端中使用以下行安装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'

为什么找不到模块？

4

1 回答 1

2

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

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

于 2021-01-20T14:25:20.457 回答