...
| 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
Simple Example for Caffe
Get node for interactive use:
| 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 |
Once on the compute node, load PowerAI module using one of theseThe HAL machine has PowerAI-1.6.0 installed for both Python2 and Python3 (default) environments. Users can load the PowerAI modules as follows:
| Code Block |
|---|
module load ibm/powerai/1.6.0.py2 # for python2 environment module load ibm/powerai/1.6.0.py3 # for python3 environment module load ibm/powerai # python3 environment by default |
Simple Example for Caffe
Initialize 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:
...