diff --git a/requirements.txt b/requirements.txt index 951a57e..8593c97 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ # Docker base image: `gcr.io/deeplearning-platform-release/tf2-gpu.2-6:latest` -tensorflow==2.6.2 # The latest should include tensorflow-gpu +tensorflow==2.12.1 # The latest should include tensorflow-gpu tensorflow-datasets==4.4.0 tensorflow-addons==0.15.0 absl-py==0.12.0