Keras
Keras is an open-source user friendly library that provides a Python interface for artificial neural networks. Keras acts as an interface for the TensorFlow library. It also supports multiple backend neural network computation.
Slurm Script
#!/bin/bash #SBATCH -J myKeras_job # Job name #SBATCH -o myKeras_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 ml load python/3.8.12-gcc-9 py-numpy/1.21.3 py-keras/2.7.0 python3.8 <ur pyKeras code>
Submit the job to the scheduler as
sbatch pyKeras.slurm
For managing your “job” , refer to this guide.
For more on Keras framework, refer to their official documentation
https://keras.io/getting_started/