Build Tensorflow in Automatic Mixed Precision Container

Hi all!

Is it possible to rebuild Tensoflow from sources in Automatic Mixed Precision Containers (eg: to speed up CPU operations and keep the FP16 compatibility?


The sources used to build the provided TF binaries are provided under /opt/tensorflow in the nvcr containers. You can tweak the sources as desired and then run /opt/tensorflow/ to compile and install a modified version of TensorFlow.