Estoy usando las siguientes líneas en la terminal para instalar rápidos y luego 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

Pero cuando voy a importar dask_cudf en el cuaderno de Python:

import dask_cudf

Aparece el siguiente error:

---------------------------------------------------------------------------
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'

¿Por qué no puede encontrar el módulo?

0
lara_toff 20 ene. 2021 a las 07:20

1 respuesta

La mejor respuesta

El comando de instalación recomendado para que Conda instale RAPIDS es el siguiente:

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

Tenga en cuenta que es posible que deba cambiar la versión del kit de herramientas de CUDA según su sistema. Puede encontrar este comando (y otras opciones de instalación) en la página Introducción.

Es posible que también desee verificar que el Python que está utilizando sea, de hecho, el del entorno Conda. Por lo general, esto se puede hacer con which python.

1
Nick Becker 20 ene. 2021 a las 14:25