内容简介：PyTorch 1.8.1 现已发布，这是一个小的错误修复版本。具体更新内容如下： New Features 改造 torch.profiler 中的 profiling tools。torch.profiler 子模块现在可用，它利用了新发布的 kineto 库进行性能分析。详...
PyTorch 1.8.1 现已发布，这是一个小的错误修复版本。具体更新内容如下：
New Features
- 改造 torch.profiler 中的 profiling tools。torch.profiler 子模块现在可用，它利用了新发布的 kineto 库进行性能分析。详情可查看博客。
- 为 pytorch xla 启用 autocast（＃48570）。该
torch.cuda.autocast软件包现在可以与 torch xla 一起使用，以提供简单的混合精度训练。
Improvements
- 使
torch.子模块导入更易于自动完成（＃52339） - 在 ONNX 中添加对
torch.{isinf,any,all}的支持（＃53529） - 在
torch.randperm的 GPU 实现中将 thrust 替换为 cub，以高提性能（＃54537）
Bug 修复
Misc
- 修复
torch.distributions验证检查（＃53763） - 允许更改
nn.Embedding的 padding 向量（＃53447） - 在 TorchScript 中正确 de-sugar Ellipsis（＃53766）
- 当组数为 24 的倍数时，停止使用 OneDNN 进行分组卷积（＃54015）
- 在 {load,store}_scalar 中使用 int8_t 代替 char（＃52616）
- 制作 ideep honor torch.set_num_thread（＃53871）
- 修复
pixel_{un}shuffle中尺寸超出范围的问题（＃54178） - 更新 kineto 以修复 libtorch 构建（＃54205）
- 修复分布式 autograd CUDA 流同步的发送/接收操作（＃54358）
ONNX
- 更新 ONNX 中的错误处理以避免
ValueError（＃53548） - 更新嵌套结构和 dict 输出的 assign output shape（＃53311）
- 更新 embedding export wrt padding_idx（＃53931）
Documentation
torch.fx的文档更新（＃53674）- 修复
distributed.rpc.options.TensorPipeRpcBackendOptions.set_device_map（＃53508） nn.LSTMCell的更新示例（＃51983）- 更新
nn.Embedding的padding_idx参数文档（＃53809） - 更新通用文档模板（＃54141）
详情可查看：https://github.com/pytorch/pytorch/releases/tag/v1.8.1
以上就是本文的全部内容，希望本文的内容对大家的学习或者工作能带来一定的帮助，也希望大家多多支持 码农网
猜你喜欢:
- NPM包（模块）发布、更新、撤销发布
- 有赞灰度发布与蓝绿发布实践
- 【重磅发布】Linkis 0.10.0 版本发布
- BeetlSQL 3.0.9 发布，Idea 插件发布
- 贝密游戏 0.7.0 发布，发布斗地主
- 【重磅发布】DataSphere Studio 0.9.0 版本发布
本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。
