The PowerAI is an enterprise software distribution that combines popular open source deep learning frameworks, efficient AI development tools, and accelerated IBM® Power Systems™ servers. The PowerAI-1.6.0 includes
Framework | Version | Description |
---|---|---|
Caffe | 1.0 | Caffe is a deep learning framework made with expression, speed, and modularity in mind. It is developed by Berkeley AI Research |
and by community contributors. | ||
TensorFlow | 1.13.1 | TensorFlow is an end-to-end open source platform for machine learning. It is developed by Google and by community contributors. |
Pytorch | 1.0.1 | Pytorch is an open source deep learning platform that provides a seamless path from research prototyping to production deployment. It is developed by Facebook and by community contributors. |
Load PowerAI Module
The HAL machine has PowerAI-1.6.0 installed for both Python2 and Python3(default) environment. Users can load the PowerAI module by input
Code Block |
---|
module load ibm/powerai/1.6.0.py2 # orfor python2 environment module load ibm/powerai/1.6.0.py3 # orfor python3 environment module load ibm/powerai # python3 environment by default |
...
Example for Caffe
Initial an interactive job
Code Block |
---|
srun --partition=debug --pty --nodes=1 --ntasks-per-node=8 --gres=gpu:v100:1 -t 01:30:00 --wait=0 --export=ALL /bin/bash |
Install samples for Caffe
Code Block |
---|
caffe-install-samples ~/caffe-samples
cd caffe-samples |
Download data for MNIST model
Code Block |
---|
./data/mnist/get_mnist.sh |
Convert data and create MNIST model
Code Block |
---|
./examples/mnist/create_mnist.sh |
Training LeNet on MNIST
Code Block |
---|
./examples/mnist/train_lenet.sh |
Simple Example for TensorFlow
...