Chainer
Chainer is a powerful, flexible and intuitive deep learning framework. It is an open source framework written purely in Python on top of NumPy and CuPy Python libraries. Chainer supports CUDA computation. It only requires a few lines of code to leverage a GPU. It also runs on multiple GPUs with little effort. Chainer supports various network architectures including feed-forward nets, convnets, recurrent nets and recursive nets. It also supports per-batch architectures.
Slurm Script
#!/bin/bash #SBATCH -J myChainer_job # Job name #SBATCH -o myChainer_job_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 python/3.8.12-gcc-9 py-numpy/1.21.3 py-chainer/7.2.0 py-matplotlib/3.5.2 python3.8 <ur pychainer code>
Submit the job to the scheduler as
sbatch pychainerjob.slurm
For managing your “job” , refer to this guide.
For more on Chainer framework, refer to their official documentation
https://docs.chainer.org/en/stable/index.html