在docker镜像中导入cuml

Question

我正在根据它的devel镜像构建一个自定义的rapidsai docker镜像。这是docker文件。

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"

但是当我用这个命令构建它的时候，

nvidia-docker build . -t test

它返回一个错误，说明：

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'

它似乎无法识别已经是基础映像库的一部分的cuml库。为什么不能导入？

Answer 1

刚刚修好了。这将不会出现任何问题。

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 "conda activate rapids" >> ~/.bashrc
SHELL ["/bin/bash", "--login", "-c"]

然后安装一些libraries..and

RUN export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu/libcuda.so.1:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
RUN export LD_LIBRARY_PATH=/usr/local/cuda-11.0/compat/libcuda.so.1:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

COPY test_cuml.py .

RUN echo "conda activate rapids" >> ~/.bashrc
SHELL ["/bin/bash", "--login", "-c"]
ENTRYPOINT ["/opt/conda/envs/rapids/bin/python", "test_cuml.py"]