在 docker 镜像中导入 cuml答案

【问题标题】：importing cuml in a docker image在 docker 镜像中导入 cuml
【发布时间】：2023-03-05 14:02:01
【问题描述】：

我正在基于其开发映像构建一个自定义的 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 库。为什么不能导入？

【问题讨论】：

标签： docker nvidia-docker

【解决方案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"]

然后安装一些库..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"]

【讨论】：