GridAI
PyTorch Lightning
Container
GridAI
PyTorch Lightning

Lightweight framework for training models at scale, without the boilerplate. Train on any number of GPUs or nodes without changing your code, and turn on advanced training optimizations with a switch of a flag.

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
["jupyter" "notebook" "--port=8888" "--no-browser" "--ip=0.0.0.0" "--allow-root"]
06/22/2021 3:56 PM UTC
sha256:978b7f5da87f071b0c87564f00fcb0282554bd12deb83257acfd7f892747aa56RUN
LIGHTNING_VERSION=1.3.7 TORCH_VERSION=$(cat torch_version.info) &&
  rm torch_version.info &&
  python --version &&
  pip --version &&
  pip list | grep torch &&
  python -c "from torch import __version__ as ver ; assert ver == '$TORCH_VERSION', ver" &&
  python -c "import pytorch_lightning as pl; print(pl.__version__)"
06/22/2021 3:56 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
PYTHONPATH=/workspace
06/22/2021 3:56 PM UTC
sha256:d09915761126ee06a3fd483df68b3c622df0e5b60cb7c9c5901f0c9b384f4ef2RUN
LIGHTNING_VERSION=1.3.7 pip install lightning-grid -U &&
  pip install "py>=1.10" "protobuf>=3.15.6" --upgrade-strategy only-if-needed
06/22/2021 3:56 PM UTC
sha256:f058410f21c3fee8c4c8955cab3629d43f68fff09f9fad79f518095620bf44ceRUN
LIGHTNING_VERSION=1.3.7 cd /workspace &&
  if [ ! -z "$LIGHTNING_VERSION" ] ; then rm -rf pytorch-lightning ; wget https://github.com/PyTorchLightning/pytorch-lightning/archive/${LIGHTNING_VERSION}.zip --progress=bar:force:noscroll ; unzip ${LIGHTNING_VERSION}.zip ; mv pytorch-lightning-*/ pytorch-lightning ; rm *.zip ; fi &&
  mv pytorch-lightning/notebooks . &&
  mv pytorch-lightning/pl_examples . &&
  python -c "fname = './pytorch-lightning/requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if not line.startswith('horovod')] ; open(fname, 'w').writelines(lines)" &&
  pip install "Pillow>=8.2" "cryptography>=3.4" "py>=1.10" --no-cache-dir --upgrade-strategy only-if-needed &&
  pip install -r ./pytorch-lightning/requirements/extra.txt --no-cache-dir --upgrade-strategy only-if-needed &&
  pip install -r ./pytorch-lightning/requirements/examples.txt --no-cache-dir --upgrade-strategy only-if-needed &&
  pip install ./pytorch-lightning --no-cache-dir &&
  rm -rf pytorch-lightning &&
  pip install jupyterlab[all] -U &&
  pip list
06/22/2021 3:55 PM UTC
sha256:6f8a774d513c4dba9c993b9f63f4b9c2a9120086ba7b62ec59c6843e073650e3COPY
dir:5f1465a607e63575f039186b80220deffde0718dbdf4444de4a4f55a74ffdc22 in /workspace/pytorch-lightning/
06/22/2021 3:55 PM UTC
sha256:97c47ace1c67e156a19dda31d1ee241ce2f536c1b6cf73340942e5c097a5af8cRUN
LIGHTNING_VERSION=1.3.7 python -c "import torch ; print(torch.__version__)" >> torch_version.info
06/22/2021 3:55 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
LIGHTNING_VERSION=
06/18/2021 7:33 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
maintainer=PyTorchLightning <https://github.com/PyTorchLightning>
06/18/2021 7:33 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=870e57ef3f56266c8d0d0cb5bdc6e4c15c39e835
04/28/2021 10:49 PM UTC
...