Dockerfile

FROM tensorflow/tensorflow:2.5.0-gpu-jupyter

RUN pip install --use-feature=2020-resolver tensorflow_hub tensorflow-probability tokenizers transformers[sentencepiece] datasets knockknock pandas panel click rich[jupyter] nltk spacy scikit-learn https://github.com/kpu/kenlm/archive/master.zip markovify

RUN pip install --use-feature=2020-resolver "jax[cuda111]" -f https://storage.googleapis.com/jax-releases/jax_releases.html flax

# RUN pip install --no-warn-script-location --use-feature=2020-resolver --user -q tensorflow_text add a check to see which version of tf it installs

# RUN pip uninstall -y tensorflow==??? TODO: automatically detect which tf-cpu version tf_text installs to be able to remove it.

CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/tf --ip 0.0.0.0 --no-browser --allow-root"]