概述【免費下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計算圖優(yōu)化、多流并行、內(nèi)存復用和模型下沉等技術(shù)手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對 PyTorch、TensorFlow 前端的友好接入能力并同時支持 onnx、pb 等主流模型格式的解析與編譯。項目地址: https://gitcode.com/cann/ge簡介本章節(jié)介紹如何使用TensorFlow的xlacompile工具將有控制流算子的網(wǎng)絡(luò)模型如圖1所示轉(zhuǎn)成函數(shù)類算子的網(wǎng)絡(luò)模型如圖2所示然后利用ATC工具轉(zhuǎn)換成適配AI處理器的離線模型。圖 1有控制流算子的網(wǎng)絡(luò)模型 絡(luò)模型?utm_sourcegitcode_repo_files)圖 2函數(shù)類算子的網(wǎng)絡(luò)模型 類算子的網(wǎng)絡(luò)模型?utm_sourcegitcode_repo_files)使用前提確保ATC工具所在服務(wù)器能連接網(wǎng)絡(luò)。安裝bazel編譯工具。請參見https://docs.bazel.build/versions/master/install-ubuntu.html官方地址安裝bazel編譯工具。安裝TensorFlow以及依賴future。ATC安裝路徑下的func2graph.py腳本依賴TensorFlow使用pip3.7.5 list查看列表中是否有TensorFlow依賴若有則不用安裝否則執(zhí)行如下命令安裝pip3.7.5 install tensorflow1.15 --userbazel編譯工具依賴python請使用pip3.7.5 list命令查看是否安裝future、patch若有則不用安裝否則執(zhí)行如下命令安裝pip3.7.5 install future --user pip3.7.5 install patch --user pip3.7.5 install numpy --user如果執(zhí)行上述命令時無法連接網(wǎng)絡(luò)且提示“Could not find a version that satisfies the requirement xxx”請參見使用pip3 install軟件時提示 Could not find a version that satisfies the requirement xxx解決?！久赓M下載鏈接】geGEGraph Engine是面向昇騰的圖編譯器和執(zhí)行器提供了計算圖優(yōu)化、多流并行、內(nèi)存復用和模型下沉等技術(shù)手段加速模型執(zhí)行效率減少模型內(nèi)存占用。 GE 提供對 PyTorch、TensorFlow 前端的友好接入能力并同時支持 onnx、pb 等主流模型格式的解析與編譯。項目地址: https://gitcode.com/cann/ge創(chuàng)作聲明：本文部分內(nèi)容由AI輔助生成（AIGC），僅供參考