/bin/sh

LIGHTNING_VERSION=1.3.8 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__)"

PYTHONPATH=/workspace

LIGHTNING_VERSION=1.3.8 pip install lightning-grid -U &&
  pip install "py>=1.10" "protobuf>=3.15.6" --upgrade-strategy only-if-needed

LIGHTNING_VERSION=1.3.8 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, !=8.3.0" "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 &&
  pip install "torchmetrics<0.4" &&
  rm -rf pytorch-lightning &&
  pip install jupyterlab[all] -U &&
  pip list

dir:fc3537f2a84f5bfe095844b2999bdd43ba39d228271d7d11d6ce5b9a4617bb7f in /workspace/pytorch-lightning/

LIGHTNING_VERSION=1.3.8 python -c "import torch ; print(torch.__version__)" >> torch_version.info

LIGHTNING_VERSION=

maintainer=PyTorchLightning <https://github.com/PyTorchLightning>

com.nvidia.build.ref=706dbb20f62dd16a38ecdf747ebc479c614b8722