Как установить tenorflow после сборки с docker - PullRequest
Новая
       59

Как установить tenorflow после сборки с docker

0 голосов
/ 07 мая 2020

Следуя руководству, найденному здесь Я успешно построил тензорный поток с моими требованиями к процессору на экземпляре Ubuntu GCP, и я с нетерпением жду возможности установить сборку на свой macbook, однако после завершения сборки нет четкого указание, как на самом деле получить установку tensorflow. У меня остался образ docker (и я не очень хорошо знаком с тем, как работает docker), вот шаги, выполняемые в руководстве:

  1. Получите найденный файл Docker здесь

Вот содержимое файла: 

FROM ubuntu:18.04

RUN apt update && apt install -y \
    build-essential \
    curl \
    git \
    wget \
    libc-ares-dev \
    libjpeg-dev \
    openjdk-8-jdk \
    gcc \
    g++ \
    python \
    python-pip \
    && rm -rf /var/lib/lists/*

ENV PYTHON_BIN_PATH="/usr/bin/python" \
    PYTHON_LIB_PATH="/usr/local/lib/python2.7/dist-packages" \
    PYTHONPATH=/tensorflow/lib \
    PYTHON_ARG=/tensorflow/lib \
    USE_BAZEL_VERSION=0.26.1 \
    TF_NEED_CUDA=0 \
    TF_NEED_GCP=0 \
    TF_CUDA_COMPUTE_CAPABILITIES=5.2,3.5 \
    TF_NEED_HDFS=0 \
    TF_NEED_OPENCL=0 \
    TF_NEED_JEMALLOC=0 \
    TF_ENABLE_XLA=0 \
    TF_NEED_VERBS=0 \
    TF_CUDA_CLANG=0 \
    TF_DOWNLOAD_CLANG=0 \
    TF_NEED_MKL=0 \
    TF_DOWNLOAD_MKL=0 \
    TF_NEED_MPI=0 \
    TF_NEED_S3=1 \
    TF_NEED_KAFKA=0 \
    TF_NEED_GDR=0 \
    TF_NEED_OPENCL_SYCL=0 \
    TF_SET_ANDROID_WORKSPACE=0 \
    TF_NEED_AWS=0 \
    TF_NEED_IGNITE=0 \
    TF_NEED_ROCM=0 \
    TF_VERSION=2.0.1\
    GCC_HOST_COMPILER_PATH="/usr/bin/gcc" \
    CC_OPT_FLAGS="-march=native"

RUN pip install pip six numpy wheel setuptools mock 'future>=0.17.1' && \
    pip install keras_applications --no-deps && \
    pip install keras_preprocessing --no-deps

RUN echo $TF_VERSION

RUN wget https://github.com/tensorflow/tensorflow/archive/v$TF_VERSION.tar.gz -q && \
    tar -xzf v$TF_VERSION.tar.gz && \
    mkdir /tensorflow && mv tensorflow-$TF_VERSION/* /tensorflow && \
    rm v$TF_VERSION.tar.gz

RUN wget https://github.com/bazelbuild/bazelisk/releases/download/v1.2.1/bazelisk-linux-amd64 -q && \
    mv bazelisk-linux-amd64 /bin/bazel && \
    chmod +x /bin/bazel

WORKDIR /tensorflow

RUN ./configure

RUN bazel build -c opt \
            --define=grpc_no_ares=true  \
            --linkopt="-lrt" \
            --linkopt="-lm" \
            --host_linkopt="-lrt" \
            --host_linkopt="-lm" \
            --action_env="LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" \
            --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both \
            --copt=-w \
            --jobs=26 \
            //tensorflow/tools/lib_package:libtensorflow
docker build -t tf .

Он продолжал работать в течение нескольких часов, а затем я получил сообщение о том, что сборка успешно завершена.

docker run -it tf /bin/bash

Что работает успешно и попадает в какую-то папку, содержащую следующее: 

root@18846955c461:/tensorflow# ls
ACKNOWLEDGMENTS     ISSUES.md          arm_compiler.BUILD  configure.cmd
ADOPTERS.md         ISSUE_TEMPLATE.md  bazel-bin           configure.py
AUTHORS             LICENSE            bazel-genfiles      models.BUILD
BUILD               README.md          bazel-out           tensorflow
CODEOWNERS          RELEASE.md         bazel-tensorflow    third_party
CODE_OF_CONDUCT.md  SECURITY.md        bazel-testlogs      tools
CONTRIBUTING.md     WORKSPACE          configure
root@18846955c461:/tensorflow#
Я получаю идентификатор изображения, используя docker ps -a

Затем я использую идентификатор изображения здесь:

docker cp image_id:/tensorflow/bazel-bin/tensorflow/tools/lib_package/libtensorflow.tar.gz tf.tar.gz

Затем я получаю небольшой tar-файл (30 мб)

Затем sudo tar -C /usr/local -xzf tf.tar.gz && sudo ldconfig

, который ничего не делает, что мне делать, чтобы продолжить?

...