This recipe shows how to run Keras using Batch AI. Keras supports tensorflow, cntk and theano backends. Currently only tensorflow and cntk backends supports running on GPU. Batch AI will automatic setup backend when toolkit is specified.
- Keras can run with CNTK or Tensorflow backend.
- Standard keras sample script mnist_cnn.py is used;
- The script downloads the standard MNIST Database on its own;
- Standard output of the job will be stored on Azure File Share.
You can find Jupyter Notebook for this recipe in Keras-GPU.ipynb.
You can find Azure CLI 2.0 instructions for this recipe in cli-instructions.md.
Under construction...
If you have any problems or questions, you can reach the Batch AI team at [email protected] or you can create an issue on GitHub.
We also welcome your contributions of additional sample notebooks, scripts, or other examples of working with Batch AI.