PyTorch 1.8.1 现已发布,这是一个小的错误修复版本。具体更新内容如下: New Features 改造 torch.profiler 中的 profiling tools。torch.profiler 子模块现在可用,它利用了新发布的 kineto 库进行性能分析。详情可查看博客。 为 pytorch xla 启用 autocast(#48570)。该torch.cuda.autocast软件包现在可以与 torch ... PyTorch 1.8.1 发布,随附 PyTorch Profiler下载地址