Skip to content

Latest commit

 

History

History
56 lines (32 loc) · 2.3 KB

README.md

File metadata and controls

56 lines (32 loc) · 2.3 KB

TOOLS

Build and run nGraph-TF bridge in Docker

To run nGraph in Docker, choose one of two ways to create your container:

  1. Use docker-build.sh to build two docker images with tags: - devel-<version> that serves as a development image which includes the prerequisites and the source - <release> that includes an installation of the bridge and itsruntime dependencies
  2. Use Dockerfile.ubuntu.18.04 by itself to set up a build environment that you can use to then manually build Tensorflow, nGraph, and the bridge in a virtualenv.
Method 1:
  • Clone the ngraph-bridge repo:

      git clone https://github.com/tensorflow/ngraph-bridge.git
    
  • Navigate into the tools directory and run the installation script:

      cd ngraph-bridge/tools
      ./docker-build.sh
    

    If you want to use build options such as --use_prebuilt_tensorflow or --use_grappler_optimizer, set an input argument when running the installation script.

      ./docker-build.sh '--use_prebuilt_tensorflow --disable_cpp_api'
    

    For more information about build options, see here. There may be some build options not supported with this method, so if your customized build is failing, Method 2 is recommended.

  • When the multi-stage docker build is complete, you will be able to run a container with Tensorflow and nGraph using the ngraph-bridge image. For instance:

      docker run -it ngraph-bridge:0.23
    

    Note: If running behind a proxy, you will need to set -e http_proxy=<http_proxy> and -e https_proxy=<https_proxy> variables in order to run the test script.

  • After running the container, you can perform an inference test by running:

      python examples/infer_image.py
    
Method 2:
  • Clone the ngraph-bridge repo:

      git clone https://github.com/tensorflow/ngraph-bridge.git
    
  • Navigate into the tools directory and build the devel image:

      cd ngraph-bridge/tools
      docker build -t ngraph-bridge:devel -f=Dockerfile.ubuntu18.04.devel .
    
  • Build the install image:

      docker build --build-arg base_image=ngraph-bridge:devel -t ngraph-bridge:master -f=Dockerfile.ubuntu18.04.install .
    
  • After the build completes, you will be able to run applications using the container:

      python examples/infer_image.py