PyTorch 1.7发布,支持CUDA 11、Windows分布式训练
机器之心报道
昨日,PyTorch 团队发布 PyTorch 1.7 版本。该版本增添了很多新特性,如支持 CUDA 11、Windows 分布式训练、增加了支持快速傅里叶变换(FFT)的新型 API 等。
CUDA 11 得到 PyTorch 官方支持;
对 autograd 分析器,更新和添加了 RPC、TorchScript 和堆栈跟踪(Stack trace)的分析和性能;
(测试版)通过 torch.fft 支持 NumPy 兼容的 FFT 操作;
(原型版)支持英伟达 A100 GPU 和原生 TF32 格式;
(原型版)支持 Windows 系统上的分布式训练。
torchvision
(稳定版)transforms 支持张量输入、批处理计算、GPU 和 TorchScript
(稳定版)JPEG 和 PNG 格式的原生图像 I/O
(测试版)新型视频读取器 API
torchaudio
(稳定版)增加了对语音录制(wav2letter)、文本转语音(WaveRNN)和源分离(ConvTasNet)的支持
添加机器之心小助手:syncedaai5,邀请加入DJL交流群。
点击阅读原文,注册直播。
© THE END
转载请联系本公众号获得授权
投稿或寻求报道:content@jiqizhixin.com
关注公众号:拾黑(shiheibook)了解更多
[广告]赞助链接:
四季很好,只要有你,文娱排行榜:https://www.yaopaiming.com/
让资讯触达的更精准有趣:https://www.0xu.cn/

随时掌握互联网精彩
赞助链接
排名
热点
搜索指数
- 1 增绿就是增优势 植树就是植未来 7945967
- 2 网购500元假茅台官方扫码为真 7990835
- 3 顾茜茜抖音账号被永久封禁 7832216
- 4 《政府工作报告》全文公布 7727396
- 5 多名外卖小哥提醒避雷黄焖鸡 7610715
- 6 夫妻连生9女 取名从招娣盼娣到仇娣 7505790
- 7 春捂要捂到什么时候?医生建议来了 7496969
- 8 老干部局招聘要求50岁 工资3000元 7329736
- 9 金秀贤把入伍期间照片私发给金赛纶 7289386
- 10 中小企业经营向暖 7167375