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"]

requirement.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。

如何在Docker中运行Tensorflow项目？

1 个答案: