【TVM 教程】编译 PyTorch 目标检测模型
原文中文,约1500字,阅读约需4分钟。发表于: 。本文介绍如何用 Relay VM 部署 PyTorch 目标检测模型。首先应安装 PyTorch。此外,还应安装 TorchVision,并将其作为模型合集(model zoo)。PyTorch 版本应该和 TorchVision 版本兼容。目前 TVM 支持 PyTorch 1.7 和 1.4,其他版本可能不稳定。
本文介绍了在Ubuntu上编译TVM的步骤,包括安装依赖项、下载源代码、编译TVM和LLVM部分等。需要注意的是,这些步骤是基于Ubuntu 18.04 LTS 64-bit环境的,其他环境可能会有所不同。