为什么 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到模块列表中Dockerfile安装，但随后的初始构建就已经失败，出现错误消息PackagesNotFoundError: The following packages are not available from current channels: - torch 。 不幸的是，遵循此处给出的建议并没有帮助我。

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

Answer 1

首先，您的假设是错误的，要验证这一点，您只需要从基础镜像运行容器，因为您可以检查官方Dockerfile或首先运行基础镜像pytorch/pytorch:latest并验证基础镜像是否正常工作你需要吗？

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

这是使用基本图像的火炬的简单示例。

至于您的 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

更新

这是使手电筒可用的方法

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