Python deep learning libraries

Central processing unit (CPU)

The following Python deep learning libraries run on CPU only, and are preinstalled for you to use in notebooks:

• TensorFlow
• Theano
• Keras
• Lasagne
• XGBoost

Graphics processing unit (GPU)

Watson Studio Local includes deep learning libraries from IBM PowerAI v1.6 which run on High Performance Computing systems with GPUs and IBM POWER9 AC922 / LC922 / LC921 servers.

PowerAI includes the following GPU enabled libraries that can be used in a Jupyter notebook with a Python 3.6 kernel:

• TensorFlow
• Pytorch
• SnapML

Additional libraries that are also supported:

• Keras

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

1. Go to Environments.
2. Find the environment Jupyter with Python 3.6 and PowerAI 1.6 for GPU for your project.
3. Click Edit settings.
4. Assign the desired number of GPUs, CPU and memory for your Jupyter container.
5. Click Save and Start. Wait until the environment is started. Note that if there are not sufficient number of GPUs available, then the environment might remain in Pending state for a long time.
6. When the environment has started, go to your project and open the desired notebook in the Jupyter with with Python 3.6 and PowerAI 1.6 for GPU environment.
7. Run !nvidia-smi in the first cell to see the following output:

   
   !nvidia-smi
   +-----------------------------------------------------------------------------+
   | NVIDIA-SMI 396.44                 Driver Version: 418.39 or higher         
             |
   |-------------------------------+----------------------+----------------------+
   | GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile
   Uncorr. ECC |
   | Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util 
   Computer M.|
   |===============================+======================+======================|
   |   0 Tesla P100-SMX2...  Off   | 00000002:01:00.0 Off |           
           0 |
   | N/A   27C   P0   29W / 300W   |       0MiB / 1628MiB |      0%   
     Default |
   +-------------------------------+----------------------+----------------------+
   
   +-----------------------------------------------------------------------------+
   | Processes:                                                      
   GPU Memory |
   |  GPU       PID  Type  Process name                              
   Usage      |
   |=============================================================================|
   |  No running processes found                                      
             |
   +-----------------------------------------------------------------------------+
   

You are now ready to run your PowerAI GPU accelerated program.