Table of contents

Python deep learning libraries

The following Python deep learning libraries are preinstalled for you to use in notebooks, depending on whether you are using Central processing unit (CPU) or Graphics processing unit (GPU):

CPU (Python 2.7/3.5) GPU PowerAI (Python 2.7) GPU (CUDA 8.0/Python 3.5) DSX Desktop
TensorFlow Y Y Y Y
Theano Y N N Y
Keras Y Y Y Y
Caffe N Y N N
XGBoost N Y N N
Lasagne Y N N N

To start a Jupyter notebook with PowerAI deep learning libraries, complete the following steps: