broadinstitute / ml4h

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tensorflow addons and probability version

StevenSong opened this issue · comments

What
tensorflow addons (and probability) v0.10.0 came out a while ago https://github.com/tensorflow/addons/releases. this version was built against TF 2.2, however we currently use the docker image for TF 2.1. because we do not version TF addons or probability, newly built docker images have dependency issues

Why
Keep dependencies up to date/synchronized

How
Set TF addons version to v0.9.1 and TF probability to v0.9.0 in tensorflow_requirements.txt or change docker image in build.sh to pull TF 2.2 docker image https://hub.docker.com/r/tensorflow/tensorflow/tags?page=1&name=2.2.0-gpu

Acceptance Criteria
new docker images are built correctly