/bin/sh

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__)"

PYTHONPATH=/workspace

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

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

dir:5f1465a607e63575f039186b80220deffde0718dbdf4444de4a4f55a74ffdc22 in /workspace/pytorch-lightning/

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

LIGHTNING_VERSION=

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

com.nvidia.build.ref=870e57ef3f56266c8d0d0cb5bdc6e4c15c39e835