升级到tensorflow 2.2

谷歌发布了TensorFlow 2.2,新版本包含字符串张量类型的升级、性能改进的Profiler、弃用SWIG转向pybind11、增强的tf.distribute功能、优化的tf.keras训练流程及SavedModel支持、TFLite新转换器和XLA在Windows上的支持等。尽管变化可能不如2.1显著,但这些改进提升了性能和用户体验。
摘要由CSDN通过智能技术生成

谷歌更新了Tensorflow 2.2。在清华的镜像网站上已经有相关的软件包。已经安装了Tensforflow的可以使用:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade tensorflow-gpu

大约需要10多分钟的时间吧。
2.2 包含了如下的升级:(来自网络)
主要特性和改进

  • 将字符串张量的标量类型从 std::string 替换为 tensorflow::tstring
  • TF 2 的新 Profiler,用于 CPU/GPU/TPU。它提供设备和主机性能分析,包括输入管道和 TF Ops。
  • 不推荐使用 SWIG,而是使用 pybind11 将 C++ 函数导出到 Python,这是弃用 Swig 所作努力的一部分。

tf.distribute:

  • 将 NVIDIA NCCL 更新到 2.5.7-1,以获得更好的性能和性能调整。
  • 支持在 float16 中减少梯度。
  • 所有实验的支持都减少了梯度压缩,以允许使用反向路径计算进行重叠梯度聚合。
  • 通过使用新添加的 tf.keras.layers.experimental.SyncBatchNormalization
    层,添加了对全局同步 BatchNormalization 的支持。该层将在参与同步训练的所有副本之间同步 BatchNormalization 统计信息。
  • 使用 tf.distribute.experimental.MultiWorkerMirroredStrategy 提高 GPU
    多工分布式培训的性能

tf.keras:

  • 可以通过覆盖 Mod
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值