TensorFlow2.1正式版发布!最后一次支持Python2,进一步支持TPU
点击我爱计算机视觉标星,更快获取CVML新技术
本文转载自机器之心。
机器之心报道
参与:杜伟、一鸣
TensorFlow2.1的更新,能够让弃坑的用户回心转意吗?
去年 10 月,谷歌才发布了 TensorFlow 2.0 正式版。时隔三个月后,昨日官方发布了 TensorFlow 2.1,本次版本更新带了了多项新特性、功能改进和 bug 修复。
从本次更新的日志来看,TensorFlow 2.1 将成为最后一个支持 Python2 的版本了。同时,本次更新的重点是增加了对 TPU 的多项支持,而 tf.keras 和 tf.data 这两个常用的 API 也得到了很多新的更新。据悉,TensorFlow 2.1 的 CUDA 版本为 10.1,cuDNN 版本为 7.6。
在对操作系统的支持上,Windows 和 Linux 系统的 TensorFlow pip 版本默认支持 GPU。也就是说,如果使用 pip install tensorflow,则版本默认为是 gpu 版本(原始的 tensorflow-gpu 版本依然存在)。当然,不管有没有英伟达版本的 GPU,tensorflow 依然能够运行。
如果需要使用 CPU 版本,用户的安装命令应该为:pip install tensorflow-cpu。
项目地址:https://github.com/tensorflow/tensorflow/releases
对于本次更新的内容,机器之心整理如下:
进一步支持 TPU
TensorFlow 2.1 最大的亮点在于进一步增加对 TPU 的支持。从 tf.keras、tf.data 等 API 的功能调整和更新来看,现在使用 TPU 加载数据集、训练和模型推理会更方便一些。
tf.keras 对 TPU 的支持
增加了在 GPU 和 Cloud TPUs 上对混合精度(mix precision)的支持;
tf.Keras 中的 compile、fit、evaluate、predict 等 API 都支持 Cloud TPUs 了,而且支持所有的 Keras 模型(即以 sequential、functional 和子类方式构建的模型都支持);
现在可以使用 Cloud TPU 进行自动外部编译(automatic outside compilation)了,这使得 tf.summary 和 Cloud TPUs 能够更好地一起使用;
分布式策略和 Keras 中的动态批大小控制已支持 Cloud TPUs;
支持在 TPU 上使用 numpy 格式的数据进行 fit、evaluate 和 predict。
tf.data 对 TPU 的支持
tf.data.Dataset 现在支持自动数据分发(automatic data distribution)和分布式环境下的分片操作,包括在 TPU pods 上都可以。
tf.distribute 对 TPU 的支持
支持在 TPU 和 TPU pods 上定制训练循环,通过以下 API 即可:
strategy.experimental_distribute_dataset、
strategy.experimental_distribute_datasets_from_function、strategy.experimental_run_v2 和 strategy.reduce。
重要 API 更新
TensorFlow 2.1 进行了以下一些重要的 API 更新:
tf.keras
推出了 TextVectorization 层,该层将原始字符串作为输入,并兼顾到了文本规范化、标记化、N 元语法生成和词汇检索;
允许 Keras .compile、.fit、.evaluate 和.predict 在 DistributionStrategy 范围之外,前提是模型在该范围内构建;
诸多流行模型的 Keras 参考实现都可以在 TensorFlow Model Garden(https://github.com/tensorflow/models/tree/master/official)中获得;
tf.data
tf.data datasets 和分布式策略都进行了改进,以获得更好的性能。需要注意的是 dataset 会变得有些不同,重新进行了分批的数据集会有多个副本;
而分布式策略也会进行调整,如下所示:
tf.data.experimental.AutoShardPolicy(OFF, AUTO, FILE, DATA)tf.data.experimental.ExternalStatePolicy(WARN, IGNORE, FAIL)
TensorRT
现在 TensorRT 6.0 是默认的版本,这一版本增加了对更多 TensorFlow 算子的支持,包括 Conv3D、Conv3DBackpropInputV2、AvgPool3D、MaxPool3D、ResizeBilinear 和 ResizeNearestNeighbor。
此外,TensorFlow 和 TensorRT 的 Python 交互 API 被命名为 tf.experimental.tensorrt.Converter。
Bug 修复
TensorFlow 2.1.0 还修复了以前版本中出现的一些 Bug,整理如下:
tf.data
修复当 sloppy=True 时 tf.data.experimental.parallel_interleave 出现的并发问题;
增加 tf.data.experimental.dense_to_ragged_batch();
扩展 tf.data 语法解析选项,从而支持 RaggedTensors。
tf.distribute
修复使用 tf.distribute.Strategy 时 GRU 崩溃或输出错误结果的问题。
tf.keras
导出 tf.keras.backend 中的 depthwise_conv2d;
在 Keras Layers 和 Models 中,删除 trainable_weights、non_trainable_weights 和 weights 中变量的重复数据;
Kerasmodel.load_weights 现将 skip_mismatch 接受为一种自变量;
修复 Keras 卷积层的输入形状缓存的行为;
Model.fit_generator、Model.evaluate_generator、Model.train_on_batch, Model.test_on_batch 和 Model.predict_on_batch 方法现遵循 run_eagerly 属性,并且在默认情况下,使用 tf.function 能够正确地运行。请注意,Model.fit_generator、Model.evaluate_generator 和 Model.predict_generator 是不宜用的端点。这些端点现已被归入支持生成器和序列的 Model.fit、Model.evaluate 和 Model.predict 中。
TF & PyTorch学习群
学习交流TensorFlow、PyTorch深度学习框架的使用,欢迎加入52CV-TF & PyTorch群,扫码添加CV君拉你入群,如已为CV君其他账号好友请直接私信,
(请务必注明:框架):
喜欢在QQ交流的童鞋可以加52CV官方QQ群:805388940。
(不会时时在线,如果没能及时通过还请见谅)
长按关注我爱计算机视觉
TensorFlow2.1正式版发布!最后一次支持Python2,进一步支持TPU相关推荐
- TensorFlow2.0正式版发布,极简安装TF2.0(CPUGPU)教程
作者 | 小宋是呢 转载自CSDN博客 [导读]TensorFlow 2.0,昨天凌晨,正式放出了2.0版本. 不少网友表示,TensorFlow 2.0比PyTorch更好用,已经准备全面转向这个新 ...
- iOS 16.1 正式版发布,苹果 App Store 上架首批支持“实时活动”应用
- Linux Kernel 5.1 正式版发布;Ubuntu 19.10 命名 Eoan Ermine
(给技术最前线加星标,每天看技术热点) 转自:开源中国.solidot.cnBeta.腾讯科技.快科技等 [技术资讯] 0.Linux Kernel 5.1 正式版发布:常规更新,非长期支持版本 大神 ...
- 『TensorFlow2.0正式版』TF2.0+Keras速成教程·零:开篇简介与环境准备
此篇教程参考自TensorFlow 2.0 + Keras Crash Course,在原文的基础上进行了适当的总结与改编,以适应于国内开发者的理解与使用,水平有限,如果写的不对的地方欢迎大家评论指出 ...
- StartOS 5.0 正式版发布
StartOS 5.0正式版发布了. StartOS -- 是由东莞瓦力网络科技有限公司发行的开源操作系统,符合国人的使用习惯,预装常用的精品软件,操作系统具有运行速度快,安全稳定,界面美观,操作简洁 ...
- XWiki 4.3 正式版发布
XWiki 4.3 正式版发布了,工作空间.扩展管理器.分发向导和 REST API 做了很多改进,改进了翻译和新的体验的 Solr 搜索. XWiki是一个由Java编写的基于LGPL协议发布的开源 ...
- 服务器安全股v4.0正式版发布 防火墙效能更强
2019独角兽企业重金招聘Python工程师标准>>> 6月24日,服务器安全狗v4.0正式版发布. 超强驱动,全线优化 服务器安全狗全方位兼容驱动V2.0震撼上线,由内而外全面提高 ...
- Xcode 7.0正式版发布了
Xcode 7.0正式版发布了 下载地址:链接: http://pan.baidu.com/s/1FNkPS 密码: ee42 本文由大学霸整理,转载请注明出处,尊重IT人!
- Wonder 1.0 正式版发布,WebGL 3D引擎和编辑器
介绍 Wonder 1.0正式版发布了.免费.开源,不用注册,直接打开在线编辑器即可使用. Wonder 是 web 端 3D 开发的解决方案,包括引擎.编辑器,致力于打造开放.分享.互助的生态.Wo ...
最新文章
- 自动安装Windows XP/自动应答,无人职守安装方法
- python中类的嵌套_python类的嵌套
- 2017《面向对象程序设计》课程作业六
- 使用python写机器学习算法遇到的问题
- rest规范 ; restful 风格; gradel介绍 ; idea安装 ;
- linux退出gnome模式,gnome-session怎么关闭比较好?
- 一键进入高通9008模式_高通3040芯片?游戏模式超低延迟?南卡lite pro全新升级!...
- 基于Android企业员工绩效考评APP的设计与实现
- 【Linux系列文章】软件包管理
- XtraReport打印二维码
- 计算机研究生学什么课本,计算机研究生到底该怎么读?
- 视频分割算法在移动端如何应用
- 物质社会会变成巨大的神经,这颗星球将会成一颗会思考的大脑
- 802.11基本元素概念介绍
- Python3 数据库(MySQL/MongoDB/Redis)
- java.sql.SQLException: Unable to run insert stmt on object 解决办法
- 人民币(RMB)小写/数字转换大写工具类,精确到分,修复‘圆零零’为‘圆整’,修复科学计数法处理异常。
- 孕期饮食新理念——初光孕妇餐 让孕期营养更科学
- win10下修改jar中的文件
- 计算机网络教程试题库,计算机网络教程_复习题及答案.doc
热门文章
- Git命令行本地库基本操作流程
- Golang笔记——json介绍
- 关于快速排序的一些理解
- F. 张胖胖玩多米诺骨牌 (南阳理工oj—21新生第二场招新赛)
- mysql alidata_linux下安装mysql | 学步园
- python sorted函数_Python 经典面试题 二
- extern 结构体_选三物质结构高中化学知识模块音频
- rip协议中周期性广播路由信息的报文_RIP路由理论知识详解(一)
- vivo手机系统升级后没有服务器,为什么安卓手机升级到了12GB还没有iPhone 4GB运行快呢?...
- 前端页面加水印插件_没用过这7款浏览器插件,你一定是假的程序员