Deep Learning software packages

IBM provides software packages for several Deep Learning frameworks and supporting libraries:

All the packages are intended for use with Ubuntu 16.04 on POWER with NVIDIA CUDA 8.0 and cuDNN v5.1 packages.

System set up

Operating System

The Deep Learning framework packages require Ubuntu 16.04 for IBM POWER8. Ubuntu installation images can be downloaded from:

http://www.ubuntu.com/download/server/power8

NVIDIA components

The Deep Learning packages require NVIDIA CUDA 8.0 and cuDNN 5.1, which can be installed as follows:

  1. Download and install NVIDIA CUDA 8.0 from https://developer.nvidia.com/cuda-downloads
  1. Download NVIDIA cuDNN 5.1 for CUDA 8.0 Power8 Deb packages from https://developer.nvidia.com/cudnn
  1. Install the cuDNN v5.1 packages

    $ sudo dpkg -i libcudnn5*deb

NVIDIA GPU driver update

The Deep Learning framework packages will work with the version 361 GPU driver that ships with CUDA 8 but IBM recommends using driver 361.93.03 or higher.

NVIDIA driver updates for POWER8 are available from https://www.nvidia.com (select DRIVERS then All NVIDIA Drivers).

Software repository Setup

The Deep Learning packages are published as an Ubuntu package that sets up an intallation repositoriy on the local machine. The repository can be enabled as follows:

  1. Download the latest mldl-repo-local .deb file from https://download.boulder.ibm.com/ibmdl/pub/software/server/mldl/

  2. Install the repository package:

    $ sudo dpkg -i mldl-repo-local*.deb
  3. Update the package cache

    $ sudo apt-get update

Tuning

Recommended settings for optimal Deep Learning performance on the S822LC for High Performance Computing:

Installing the Deep Learning Frameworks

Installing all frameworks at once

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

    $ sudo apt-get install power-mldl

Installing frameworks individually

The Deep Learning frameworks can be installed individually if preferred. The framework packages are:

Each can be installed with:

    $ sudo apt-get install <framework>

Installation notes

Framework installation may pull in just a few or many (up to a few hundred) pre-requisite packages depending on currently installed packages.

Getting started with MLDL Frameworks

General setup

Each framework package provides a shell script to simplify environmental setup.

We recommend that users update their shell rc file (e.g. .bashrc) to source the desired setup scripts. For example:

    source /opt/DL/<framework>/bin/<framework>-activate

Each frame also provides a test script to verify basic function:

    $ <framework>-test

Getting started with Caffe

BVLC and NVIDIA Caffe alternatives

Packages for both upstream BVLC Caffe (/opt/DL/caffe-bvlc), IBM optimized BVLC Caffe (/opt/DL/caffe-ibm), and NVIDIA's fork of Caffe (/opt/DL/caffe-nv) are provided. You can choose which of those packages is the system default (/opt/DL/caffe) using Ubuntu's alternatives system:

    $ sudo update-alternatives --config caffe
    There are 3 choices for the alternative caffe (providing /opt/DL/caffe).

      Selection    Path                Priority   Status
    ------------------------------------------------------------
    * 0            /opt/DL/caffe-ibm   100        auto mode
      1            /opt/DL/caffe-bvlc   50        manual mode
      1            /opt/DL/caffe-ibm   100        manual mode
      2            /opt/DL/caffe-nv     75        manual mode

    Press enter to keep the current choice[*], or type selection number:

Users can activate the system default caffe:

    source /opt/DL/caffe/bin/caffe-activate

Or they can activate a specific variant. For example:

    source /opt/DL/caffe-bvlc/bin/caffe-activate

Attempting to activate both Caffe packages in a single login session will cause unpredictable behavior.

Caffe samples and examples

Each Caffe package includes example scripts and sample models, etc. A script is provided to copy the sample content into a specified directory:

    $ caffe-install-samples <somedir>

More info

Visit Caffe's website at http://caffe.berkeleyvision.org/ for tutorials and example programs that you can run to get started.

Here are links to a couple of the example programs from Caffe's website:

Getting started with Theano

Here are some links to help you get started with Theano:

Visit Theano's website for the latest from Theano.

Getting started with Torch

The Torch Cheatsheet contains lots of info for people new to Torch, including tutorials and examples.

Here is a Torch tutorial that includes the code for several example programs:

Visit Torch's website for the latest from Torch.