Skip to content

Installing MaTEx TensorFlow CPU

Vinay Amatya edited this page Jun 27, 2017 · 16 revisions

We have provided a set of scripts for easy installation on CPU clusters. We expect that minimal changes would be required to the script for installation.

For using CPU enabled TensorFlow, you would need to set up the MPI_HOME environment variable to point where MPI and should be found.

As an example, for bash shells:

$ export MPI_HOME=/where/mpi_library/is/installed

Installation for bash shells

$ cd matex/src/deeplearning/tensorflow/cpu/py3.x
$ source ./install_mpi_tf.sh

Installation for C-shells

$ cd matex/src/deeplearning/tensorflow/cpu/py3.x
$ source ./install_mpi_tf.csh

This ensures that you are in a virtual python environment that encapsulates the distributed TensorFlow changes. Your shell prompt should look differently.

[py_distro] $
Clone this wiki locally