importing cuml in a docker image

  docker, nvidia-docker

I’m building a custom rapidsai docker image based on its devel image. Here is the docker file.

FROM rapidsai/rapidsai-dev:0.19-cuda11.0-devel-ubuntu18.04-py3.7

# Defining working directory and adding source code
WORKDIR /usr/src/app

RUN echo "Make sure cuml is installed:"
RUN python -c "import cuml"

But it returns an error saying:

Step 4/4 : RUN python -c "import cuml"
 ---> Running in 553d12bf7e68
Traceback (most recent call last):
  File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'cuml'

It seems that it can’t recognize cuml library which is already a part of the libraries of the base image. Why it can’t import it?

Source: Docker Questions

LEAVE A COMMENT