MxNet

Apache MXNet

Apache MXNet is an open-source deep learning software 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 programming languages (including C++, Python, Java, Julia, MATLAB, JavaScript, Go, R, Scala, Perl, and Wolfram Language). The MXNet library is portable and can scale to multiple GPUs as well as multiple machines.

Slurm Script

#!/bin/bash
#SBATCH -J mymx_codejob # Job name
#SBATCH -o mymx_code_output.%j.out # Name of stdout output file (%j expands to jobId)
#SBATCH -p CUIQue # Queue name
#SBATCH -N 1 # Total number of nodes requested
#SBATCH -n 16 # Total number of mpi tasks requested

module load mxnet/1.8.0 python/3.8.12 py-numpy/1.22.4
python3.8 mymxnet.py

mymxnet.py Code

import mxnet as mx
a = mx.nd.ones((2, 3))
b = a * 2 + 1
b.asnumpy()
print(b.asnumpy())

Submit the job to the scheduler as

sbatch mymx.slurm

For managing your “job” , refer to this guide.

For more on MxNet, refer to their official documentation
https://github.com/apache/mxnet