Installing the deep learning frameworks

Find instructions for installing the deep learning framework on an IBM® Power® System server.

Setting up the software repository

The PowerAI deep learning packages are distributed in a tar.gz file that contains an rpm and a readme file. The tar.gz file must be extracted on the local system. Installing the rpm creates an installation repository on the local system.

Install the repository package by running the following command:

sudo rpm -ihv mldl-repo-*.rpm

Installing all frameworks at the same time

All the Deep Learning frameworks can be installed at the same time by using the power-mldl meta-package:

sudo yum install power-mldl
Note: This step does not include installing PowerAI distributed deep learning (DDL) packages. See details of how to install DDL packages.

Installing the Python 3 versions of the frameworks

The Python 3 versions of the frameworks can be installed at the same time by using the power-mldl-py3 meta-package.

sudo yum install power-mldl-py3

Installing frameworks individually

You can install the deep learning frameworks individually. The framework packages include the following versions:

Table 1. Framework packages
Package Description Version
caffe-bvlc Berkeley Vision and Learning Center (BVLC) upstream Caffe 1.0.0
caffe-ibm IBM Optimized version of BVLC Caffe 1.0.0
pytorch PyTorch 0.4.1
tensorflow TensorFlow 1.10.0
tensorboard Web Applications for inspecting TensorFlow runs and graphs. 1.10.0

The Python 3 version of each framework appends -py3 to the package name.

Table 2. Python 3 versions
Package Description Version
pytorch-py3 PyTorch 0.4.1
tensorflow-py3 TensorFlow 1.10.0
tensorboard-py3 Web Applications for inspecting TensorFlow runs and graphs. 1.10.0

To install each of these frameworks, run the following command:

sudo yum install <framework>

Installing IBM PowerAI distributed deep learning (DDL) packages

To enable distributing model training across a cluster of Power Systems™ servers, install PowerAI distributed deep learning. DDL includes IBM Spectrum™ MPI for communication among systems.

Install the PowerAI distributed deep learning packages by using the following command:

sudo yum install power-ddl
Note: DDL is an optional component. Other PowerAI components can be installed and used without installing DDL.

To use InfiniBand for DDL communications, install the latest Mellanox OFED driver. See the Download tab at: http://www.mellanox.com/page/products_dyn?product_family=26

Installing IBM PowerAI Snap ML packages

Install the PowerAI Snap ML packages by using the following command:

sudo yum install power-snapml
Note: Snap ML is an optional component. Other PowerAI components can be installed and used without installing Snap ML.

Accepting the PowerAI license agreement

Read the license agreement and accept the terms and conditions before you use any of the frameworks.

sudo /opt/DL/license/bin/accept-powerai-license.sh

After you finish reading the license agreement, future installs can be automated to silently accept the license agreement.

sudo IBM_POWERAI_LICENSE_ACCEPT=yes /opt/DL/license/bin/accept-powerai-license.sh