ibmcom/powerai

By ibmcom

Updated over 5 years ago

Official docker images for IBM PowerAI

Image
33

100K+

ibmcom/powerai repository overview

Requirements

PowerAI is optimized to leverage the unique capabilities of IBM Power Systems accelerated servers and x86_64 platform servers. It is supported on:

  • IBM AC922 POWER9 system with NVIDIA Tesla V100 GPUs
  • IBM S822LC POWER8 system with NVIDIA Tesla P100 GPUs
  • IBM Power System S822LC with NVIDIA Tesla P100 GPUs
  • x86_64 systems with NVIDIA Tesla V100, P100, or T4 GPUs

Host System Requirements

ComponentVersion
Red Hat7.6
Ubuntu18.04
Docker1.13.1
NVIDIA Docker2.0*
NVIDIA GPU driver440

*All PowerAI images starting with 1.5.4 require nvidia-docker 2.0

  • Configuring host for PowerAI

    To setup your machine for use with PowerAI please follow instructions described in IBM's Knowledge Center for Red Hat or Ubuntu here

    The instructions include how to install the NVIDIA GPU driver, docker, nvidia-docker etc.

Using the PowerAI image from Docker Hub

To start up a PowerAI container.

Red Hat docker
docker run -ti --env LICENSE=yes ibmcom/powerai:<tag> bash 
docker-ce
nvidia-docker run -ti --env LICENSE=yes ibmcom/powerai:<tag> bash 

  PyTorch Users: If you plan on using any multiprocessor data loader with PyTorch. The default shared memory segment size for the container may not be large enough. You can increase the shared memory size with either --ipc=host or --shm-size command line options on docker run

You can read more about this issue on PyTorch's Readme https://github.com/pytorch/pytorch/blob/master/README.md under the "Docker image" section.


License Acceptance

You must accept the licenses of all included components before using a PowerAI container. View the WML CE license locally in the image at $HOME/.powerai/powerai-license/1.7.0/license# or View the WML CE license externally at https://github.com/IBM/powerai/tree/master/containers/1.7.0

  • Accept the licenses at docker runtime by adding the --env LICENSE=yes parameter on the docker run command line

or

  • Accept the license within an already running container accept-powerai-license.sh

Tag Formats

  • latest - PowerAI 1.7.0 and Anaconda for python 3.7

  • <powerai-version>-<framework>-<cpu>-<OS>-<python>-<architecture>

    • powerai-version - The version of PowerAI installed in image latest is 1.7.0
      Available options - 1.5.2, 1.5.3, 1.5.4, 1.6.0, 1.6.1 1.6.2 1.7.0

    • framework - The framework installed on the image, (options vary depending on image version)
      Available options - all, tensorflow(>=1.6.0), tensorflow-serving (>=1.6.1), pytorch(>=1.6.0), caffe(>=1.6.0), snap-ml(>=1.6.0, ppc64le only, >=1.7.0, x86_64 and ppc64le), xgboost(>=1.6.1, ppc64le only, >=1.6.2, x86_64 and ppc64le), rapids (>=1.6.2, ppc64le only)

    • cpu - Starting with PowerAI 1.6.1, additional images are built with cpu only versions of deep learning frameworks. These versions have the -cpu flag appended to the framework name.
      Available options - all-cpu(>=1.6.1), tensorflow-cpu(>=1.6.1), caffe-cpu(>=1.6.1), xgboost-cpu(>=1.6.1, ppc64le only, >=1.6.2, x86_64 and ppc64le), pytorch-cpu(>=1.6.2), tensorflow-serving-cpu(>=1.7.0)

    • os - The operating system installed on the image (options vary depending on image version)
      Available options - ubuntu16.04(<1.6.0), ubuntu18.04(>=1.6.0)

    • python - The python version used by frameworks
      Available options - py3(<=1.6.1), <none>(python2, (<=1.6.1)), py36(>=1.6.2), py37 (>=1.6.2)

    • architecture - Starting with PowerAI 1.6.0, images are built to support x86_64 and ppc64le architectures (options vary depending on image version)
      Available options - ppc64le(>=1.6.0), x86_64(>=1.6.0), <none>(>=1.6.0 docker will auto detect on your box, <1.6.0 docker will serve ppc64le versions)

    examples

    1.7.0-pytorch-ubuntu18.04-py36 #Download pytorch image for the requesting machine's architecture with python 3.6

    1.7.0-all-ubuntu18.04-py37 #Download all frameworks image for the requesting machine's architecture with python 3.7


Installed Packages

PowerAI provides software packages for several Deep Learning frameworks, supporting libraries, and tools:

Component1.5.2 Images1.5.3 Images1.5.4 Images1.6.0 Images1.6.1 Images1.6.2 Images1.7.0 Images
Distributed Deep Learning (DDL)1.0.01.1.01.2.01.3.01.4.01.5.01.5.1
TensorFlow1.8.01.10.01.12.01.13.11.14.01.15.02.1.0
TensorFlow ProbabilityNANA0.5.00.6.00.7.00.8.00.9.0
TensorFlow EstimatorNANANA1.13.01.14.01.15.12.1.0
TensorFlow ServingNANANANA1.14.01.15.02.1.0
TensorRTNANANANA5.1.3.66.0.1.57.0.0.11
TensorBoard1.8.01.10.01.12.01.13.01.14.01.15.02.1.0
IBM Caffe1.0.01.0.01.0.01.0.01.0.01.0_1.6.21.0_1.7.0
BVLC Caffe1.0.01.0.01.0.0NANANANA
Caffe2NANA1.0rc11.0.11.1.01.2.01.3.1
snap-ml1.0.01.0.0NANANANA1.6.0
snapml-sparkNANA1.0.01.2.01.3.01.4.01.6.0
pai4skNANA1.0.01.3.01.4.01.5.01.6.0
xgboostNANANANA0.820.900.90
Spectrum MPI10.210.210.210.210.0310.0310.03
OpenBLAS0.2.200.3.20.3.30.2.200.2.200.3.60.3.6
Protobuf3.4.03.4.03.6.13.6.13.7.13.8.03.8.0
ONNXNANA1.3.01.3.01.5.01.5.01.6.0
Rapids cuDFNANANA0.2.00.7.20.9.00.11.0
Rapids cuMLNANANA0.2.00.7.00.9.10.11.0
apexNANANANANA0.1.0_1.6.20.1.0_1.7.0
daskNANANANANA2.3.02.9.2
dask-xgboostNANANANANA0.1.70.1.9
arrow-cppNANANANA0.12.10.14.10.15.1
horovodNANANANANANA0.19.0
pytorch0.4.00.4.11.0rc11.0.11.1.01.2.01.3.1
CUDA9.2.889.2.14810.0.13010.110.110.110.2
cuDNN7.1.47.2.17.3.17.57.57.6.37.6.5
NCCL2.2.122.2.132.3.52.4.22.4.72.4.82.5.6
conda4.5.114.5.114.5.114.5.114.6.144.7.124.8.1
Ubuntu16.0416.0418.0418.0418.0418.0418.04

Container Usage

User Ids

To satisfy our cloud users, and to stay inline with the principle of least privilege, the default user is pwrai in the images.

pwrai has a uid:gid of 2051:2051 and has password-less sudo setup.

For those that wish to use root, it is enabled, and available via the docker --user root argument at runtime.

For security reasons, we recommend you take advantage of Docker namespaces as documented here

Running PowerAI Frameworks

Please reference the "Getting Started with MLDL Frameworks" page here

Using TensorFlow-Serving container

The PowerAI TensorFlow-Serving docker container is modeled after the tensorflow/serving docker container provided by TensorFlow. As such all TensorFlow documentation for TensorFlow Serving with Docker applies to the PowerAI TensorFlow Serving image. Just replace tensorflow/serving in all examples with ibmcom/powerai:<powerai-version>-tensorflow-serving-ubuntu18.04-<py36/py37> or ibmcom/powerai:<powerai-version>-tensorflow-serving-cpu-ubuntu18.04-<py36/py37>


LICENSES


NVIDIA

  • CUDA Toolkit To view the license for the CUDA Toolkit included in this image, click here

  • CUDA Deep Neural Network library (cuDNN) To view the license for cuDNN included in this image, click here


Anaconda

The Anaconda User's license can be viewed at (https://docs.anaconda.com/anaconda/eula)

The list of installed python packages under anaconda can be displayed using pip show <packagename> | grep License:

To view a python package's specific License, go to the package's website displayed by the pip show <packagename> | grep Home-page:


Ubuntu

Ubuntu's(Canonical) Legal information can be viewed at (https://www.ubuntu.com/legal)

The list of installed Debian packages can be seen using dpkg --list

The license of a particular Debian package can be viewed inside the PowerAI image under /usr/share/<packagename>/copyright


PowerAI

View the WML CE license locally in the image at $HOME/.powerai/powerai-license/1.7.0/license# or View the WML CE license externally at https://github.com/IBM/powerai/tree/master/containers/1.7.0

Tag summary

Content type

Image

Digest

Size

6.1 GB

Last updated

over 5 years ago

docker pull ibmcom/powerai:1.6.2-all-ubuntu18.04-py37-ppc64le