docker - 如何在Docker中运行Tensorflow项目？

Question

我是Docker的新手，但是我已经搜索了很多有关我面临的问题的信息。

我在使用Tensorflow，PyQt和其他软件包的代码中。现在，我提取了tensorflow/tensorflow:1.4.0-gpu-py3和nvidia/cuda:8.0-cudnn6-runtime。另外，我还构建了具有某些依赖项的应用程序镜像。

我试图使用docker-compose运行以上所有图像，如下所示:

version: '3'
services:
  nvidia:
    image: "nvidia/cuda:8.0-cudnn6-runtime"

  tensorflow:
    image: "tensorflow/tensorflow:1.4.0-gpu-py3"

  app:
    image: my_app

但是我收到错误 ImportError: No module named 'tensorflow'。

请提出建议解决方法，以帮助我。

编辑:

以下代码示例仅是我代码的几行。

import sys
from PyQt5 import QtCore, QtGui, QtQml, QtQuick
from OpenGL import GL
import cv2 # .cv2 as cv2
from multiprocessing import Process,Queue, Value, Manager
import os
import tensorflow as tf

编辑:

# Use an official Python runtime as a parent image
FROM ubuntu:16.04

# Set the working directory to /app
WORKDIR /app

# Copy the current directory contents into the container at /app
COPY . /app

RUN \
  apt-get update && \
  apt-get install -y python python-dev python-pip python-virtualenv && \
  rm -rf /var/lib/apt/lists/*


RUN apt-get update && apt-get install -y --no-install-recommends apt-utils && apt-get install -y libgtk2.0-dev python python-dev python3 python3-dev python3-pip

RUN apt-get update && apt-get install -y build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev

RUN pip install setuptools pip --upgrade --force-reinstall


# Install any needed packages specified in requirements.txt
RUN pip install --trusted-host pypi.python.org -r requirements.txt

#RUN apt-get update -y

# Install packages
#RUN apt-get install -y curl
#RUN apt-get install -y postgresql
#RUN apt-get install -y postgresql-client
#RUN apt-get install -y python3-numpy python3-opengl python-qt4 python-qt4-gl


# Run app.py when the container launches
CMD ["python3", "Working.py"]

required.txt

PyOpenGL
PyQt5
opencv-python

Answer 1

您有3个单独的Docker容器，Nvidia，Tensorflow和您的应用程序。

当您在python应用程序中包含tensorflow时，那里没有Tensorflow包，它位于单独的容器中。

建议删除Tensor-flow容器，然后将应用程序添加到tensorflow图像中。

在您的Dockerfile中更改镜像:

FROM ubuntu:16.04 to FROM tensorflow/tensorflow:1.4.0-gpu-py3

然后更改Dockerfile安装的其他部分，因为tensorflow镜像已经安装了python3。