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 中法关系如何“走得更稳更好” 7904684
- 2 日方军机滋扰擅闯或被视为训练靶标 7808635
- 3 马斯克公开呼吁:废除欧盟 7713895
- 4 今年流感为啥这么“凶” 7617765
- 5 8.9元抗炎神饮被年轻人抢断货 7520097
- 6 新规让“幽灵外卖”现原形 7423668
- 7 男子海洋馆内抽烟被白鲸喷水浇灭 7330017
- 8 国乒对阵日本 双方名单出炉 7238636
- 9 五粮液降价到800多元?公司回应 7140780
- 10 2000多年前中国水利设计有多超前 7048314







机器之心
