極智AI | ubuntu編譯安裝onnx-mlir的正确姿勢

歡迎關注我的公衆号 [極智視界]，擷取我的更多經驗分享

大家好，我是極智視界，本文介紹一下 ubuntu編譯安裝onnx-mlir的正确姿勢。

邀您加入我的知識星球「極智視界」，星球内有超多好玩的項目實戰源碼下載下傳，連結：https://t.zsxq.com/0aiNxERDq

首先需要說說：什麼是MLIR，什麼是ONNX-MLIR。ONNX-MLIR是一個基于MLIR（Multi-Level Intermediate Representation）的開源項目，旨在将ONNX（Open Neural Network Exchange）模型轉換為MLIR IR，以實作高效、可擴充和可優化的機器學習推理。而MLIR是一種多級别中間表示（IR），它提供了一個統一的架構，用于将各種不同的前端語言和後端目标之間的高效、可重用和可擴充的轉換。ONNX-MLIR将ONNX圖轉換為MLIR IR，使得可以使用MLIR所提供的一些進階優化技術，例如優化緩沖區配置設定、運作時縮減和特定硬體的代碼生成。

這裡咱們來看看在ubuntu上如何正确編譯安裝onnx-mlir。

下面開始。

一、建立容器

# 首先拉取鏡像
docker pull nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04