PyTorch Vision项目支持Python 3.12的适配工作

随着Python 3.12的发布，PyTorch Vision团队积极开展了对新版本Python的支持工作。本文将详细介绍这一适配过程中的技术挑战和解决方案。

背景与挑战

Python 3.12引入了一些语法和内部实现的变更，这给PyTorch Vision这样的深度学习计算机视觉库带来了适配需求。主要挑战来自于Python 3.12中AST(抽象语法树)的修改，这些修改导致PyTorch在运行时产生了一些新的弃用警告。

关键问题分析

在测试过程中，团队发现transform模块的测试用例会因严格的警告捕获机制而失败。具体来说，Python 3.12的AST变更触发了PyTorch内部的弃用警告，而这些警告在测试环境中被配置为视为错误处理。

解决方案

团队采用了多层次的解决方案：

1. 测试配置调整：修改了测试文件中的警告过滤器配置，在保持错误级别警告捕获的同时，特别忽略来自Python 3.12的DeprecationWarning。

2. 版本兼容性规划：在支持新版本的同时，团队也规划了旧版本Python的支持策略，确保平稳过渡。

实施细节

在transform模块的测试文件中，原本单一的警告过滤器配置被扩展为包含两个标记：

• 保持原有的错误级别警告捕获
• 新增对DeprecationWarning的忽略规则

这种细粒度的控制既保证了测试的严格性，又避免了因Python版本更新带来的误报。

后续工作

完成Python 3.12支持后，团队将继续关注：

• PyTorch核心库对Python 3.12的完全适配进展
• 其他可能受影响的模块和功能
• 性能优化和稳定性提升

这次适配工作展示了PyTorch Vision团队对保持技术前沿性的承诺，也为用户提供了在新Python版本中使用最新计算机视觉功能的可能性。