为什么 pytorch Docker 镜像没有自带torch？

Question

我正在尝试构建一个 docker 容器，使我能够运行需要 python torch 模块的代码。

我选择从pytorch/pytorch:latest 基础镜像开始我的镜像并手动添加一些必需的模块。构建、推送和拉取到远程服务器成功（至少没有错误消息）。

目前我的 Dockerfile 看起来像这样：

FROM pytorch/pytorch:latest

RUN apt-get update \
     && apt-get install -y \
        libgl1-mesa-glx \
        libx11-xcb1 \
     && apt-get clean all \
     && rm -r /var/lib/apt/lists/*

RUN /opt/conda/bin/conda install --yes \
    astropy \
    matplotlib \
    pandas \
    glob2 \
    PIL \
    scikit-learn \
    scikit-image \
    numpy

但是，在容器中运行我的 python 脚本时，我得到ImportError: No module named torch。这让我觉得很奇怪，因为它让我假设 pytorch 基础映像不包含 torch 模块...？

尽管如此，我还是尝试将 torch 添加到要在 Dockerfile 中安装的模块列表中，但是初始构建已经失败并出现错误消息 PackagesNotFoundError: The following packages are not available from current channels: - torch。不幸的是，遵循here 的建议对我没有帮助。

我很感激任何解释为什么在构建的容器中找不到 torch 模块，当然还有任何帮助来解决这个问题！谢谢！

Answer 1

首先，您的假设是错误的，要验证这一点，您只需从基础镜像运行容器，您可以查看官方的Dockerfile 或首先运行基础镜像 pytorch/pytorch：最新 并验证基础映像是否按您的需要工作？

这是官方镜像中已安装模块的列表，在列表底部，您可以看到火炬。

这是一个使用基础图像的 torch 的简单示例。

至于您的 Dockerfile，所以包 PIL 正在从头开始破坏 docker 构建，但如果 PyTorch 是基础映像，则这不可见。

由于某种原因，我在子图像中找不到火炬，所以使用 pip install 安装它，然后它就可以工作了。

这里是Dockerfile：

FROM pytorch/pytorch:latest

RUN apt-get update \
     && apt-get install -y \
        libgl1-mesa-glx \
        libx11-xcb1 \
     && apt-get clean all \
     && rm -r /var/lib/apt/lists/*

RUN /opt/conda/bin/conda install --yes \
    astropy \
    matplotlib \
    pandas \
    glob2 \
    scikit-learn \
    scikit-image \
    numpy \ 
    torch

更新

这里是让torch可用的方法

FROM pytorch/pytorch:latest

RUN apt-get update \
     && apt-get install -y \
        libgl1-mesa-glx \
        libx11-xcb1 \
     && apt-get clean all \
     && rm -r /var/lib/apt/lists/*

RUN /opt/conda/bin/conda install --yes \
    astropy \
    matplotlib \
    pandas \
    scikit-learn \
    scikit-image 

RUN pip install torch