我只是想知道为什么tensorflow/tfx图像是8Gb,而gpu tensorflow/tensorflow图像小于3gB?在运行时,我希望tfx容器只是加载输入工件,运行一些tensorflow代码,然后保存输出工件。所以我不能算出权重
Thanks and Regards
Youcef
这里有一些解释为什么它这么重:
自0.28以来实际发生了增加,因为我们现在基于一个经过验证的构建,它与GPU开箱即用兼容。
你也可以在python的基本镜像上构建自己的镜像。