Skip to content

Latest commit

 

History

History
 
 

popdist

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
test
test
 
 
README.md
README.md
 
 
popdist_training.py
popdist_training.py
 
 

README.md

Graphcore

PopDist example

PopDist (Poplar Distributed Configuration) provides a set of APIs which can be used to write a distributed application. The application can then be launched on multiple instances through PopRun, our command line utility.

This example contains a TensorFlow CNN with PopDist support, which can be launched on multiple instances using a PopRun command line.

You can learn more about PopDist and PopRun in the PopDist and PopRun User Guide.

File structure

  • popdist_training.py Example training script with PopDist support.
  • test Integration tests for this example.
  • README.md This file.

How to use this demo

  1. Prepare the TensorFlow environment.

    Install the Poplar SDK following the instructions in the Getting Started guide for your IPU system. Make sure to source the enable.sh script for Poplar and activate a Python virtualenv with either a TensorFlow 1 or TensorFlow 2 wheel from the Poplar SDK installed (use the version appropriate to your operating system and processor).

  2. Run the script using PopRun. The number of instances and replicas are provided as command-line arguments. Example:

    poprun --num-instances=2 --num-replicas=4 python3 popdist_training.py