MXnet

Apache MXNet is a modern open-source deep learning framework used to train, and deploy deep neural networks. It is scalable, allowing for fast model training, and supports a flexible programming model and multiple languages (C++, Python, Julia, Matlab, JavaScript, Go, R, Scala, Perl, Wolfram Language)

MXnet has been compiled and installed for the stock installation of Python and the Python 3.6.1 module.

[gqd693@login-0-0 ~]$ qlogin -q gpu.q
local configuration login-0-0.cm.cluster not defined - using global configuration
Your job 70000 ("QLOGIN") has been submitted
waiting for interactive job to be scheduled ...
Your interactive job 70000 has been successfully scheduled.
Establishing /cm/shared/apps/sge/var/cm/qlogin_wrapper session to host gpu02.cm.cluster ...
Last login: Thu May 3 10:41:40 2018 from login-0-0.cm.cluster

[gqd693@gpu02 ~]$ module load mxnet
[gqd693@gpu02 ~]$ python
Python 2.7.5 (default, Aug 4 2017, 00:39:18)
[GCC 4.8.5 20150623 (Red Hat 4.8.5-16)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import mxnet as mx
>>> a = mx.nd.ones((2, 3), mx.cpu())
>>> print ((a * 2).asnumpy())
[[2. 2. 2.]
 [2. 2. 2.]]
>>> a = mx.nd.ones((2, 3), mx.gpu())
>>> print ((a * 2),asnumpy())
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
NameError: name 'asnumpy' is not defined
>>> print ((a * 2).asnumpy())
[[2. 2. 2.]
 [2. 2. 2.]]
>>>

Training Examples

[gqd693@gpu01 ~]$ mkdir mxnet-examples
[gqd693@gpu01 ~]$ cd mxnet-examples/
[gqd693@gpu01 mxnet-examples]$ rsync -avP /apps/mxnet/example/ .
[gqd693@gpu01 mxnet-examples]$ cd image-classification/
[gqd693@gpu01 mxnet-examples]$ module load mxnet
[gqd693@gpu01 image-classification]$ python train_mnist.py --network mlp --gpus 0

-- AdminUser - 03 May 2018