使用torch.compile()遇RuntimeError：Python3.12+不支持Dynamo求解决

报错原因 #

使用torch.compile()时触发错误，核心原因是Python 3.12及以上版本暂不支持PyTorch的Dynamo组件，而torch.compile()依赖该组件实现模型编译优化。

解决方案 #

• 降级Python版本到3.11或更低
  卸载当前Python 3.12，安装Python 3.11版本，重新配置环境后执行pip install torch torchvision torchaudio，安装适配版本的PyTorch。
• 临时跳过模型编译步骤
  直接注释掉编译相关代码，或把控制编译的compile变量设为False，示例：

  # if compile:
  #     print("compiling the model")
  #     unoptimized_model = model
  #     model = torch.compile(model)
  

• 等待PyTorch官方更新
  PyTorch后续版本会逐步适配Python 3.12+的Dynamo支持，可关注官方版本更新日志。

内容的提问来源于stack exchange，提问作者Astolsko