↓Skip to main content

MNIST training starter kit with Ray, Lightning and PyTorch

·
Deep Learning Jupyter Machine Learning MLops

Some time ago, I shared a starter kit for reinforcement learning research with Ray and RLlib. The following is a productive tech stack for doing deep learning research.

What makes this stack challenging to use is that the constituent packages are updated frequently. In my experience, if I just install the latest versions of each package, sometimes the stack runs out-of-the-box and sometimes not.

Here, I share a working snapshot of the tech stack. There is an example MNIST training program and a requirements.txt for reproducing the run-time environment.

The source code is at this repository: https://github.com/rlan/ray-pytorch-lightning

You might also be interested in:

  • a runtime starter kit based in docker for running PyTorch and TensorFlow code on CPU or GPU.