search

tensorflow / tpu

Reference models and tools for Cloud TPUs.
https://cloud.google.com/tpu/
Apache License 2.0
5.2k stars 1.77k forks source link

Bug in r.27 release tpu/tools/docker/Dockerfile #1022

Closed danielvegamyhre closed 11 months ago

danielvegamyhre commented 11 months ago

TensorFlow installation is using v2.2.0: https://github.com/tensorflow/tpu/blob/r2.7/tools/docker/Dockerfile#L27

However, it should be 2.7 since tensorlfow/models and tensorflow/tpu are both using 2.7: https://github.com/tensorflow/tpu/blob/r2.7/tools/docker/Dockerfile#L33-L36

danielvegamyhre commented 11 months ago

cc @aman2930

aman2930 commented 11 months ago

Fixed it.