近日，MegEngine 在 v1.13.3 正式完成了與國(guó)內(nèi) AI 芯片獨(dú)角獸寒武紀(jì)旗下云端人工智能處理器芯片的兼容性適配。MegEngine 與 寒武紀(jì)目前完成了常見 cv 模型的算子適配，這也標(biāo)志著寒武紀(jì)端云一體的人工智能芯片，與 MegEngine 深度學(xué)習(xí)框架的成功融合。
如何在寒武紀(jì)上運(yùn)行 MegEngine
MegEngine 寒武紀(jì)安裝說明
- 安裝的前置條件：
- 安裝寒武紀(jì) MLU 驅(qū)動(dòng)，sdk-1.13.0-driver
- 安裝 cntoolkit, cnnl, cnnl_extra, cncl, cnlight
- 安裝 MagicMind
- 安裝方式：通過源碼編譯安裝
- 下載源碼&編譯
git clone https://github.com/MegEngine/MegEngine.git megengine
cd megengine
# 構(gòu)建用于編譯并打包的 docker 鏡像
./scripts/whl/manylinux2014/build_image.sh
# 下載 mkl 依賴
./third_party/install-mkl.sh
# 編譯并打包
export ALL_PYTHON="39" # 指定編譯的 python 版本，可選 36m 37m 38 39 310
export NEUWARE_HOME=/usr/local/neuware # 指定寒武紀(jì) sdk 安裝路徑
./scripts/whl/manylinux2014/build_wheel_common.sh -sdk neuware113
- 安裝與驗(yàn)證生成的whl包
- 安裝生成的 whl 包，并進(jìn)行測(cè)試
cd megngine
pip3 install ./scripts/whl/manylinux2014/output/wheelhouse/neuware113/MegEngine-1.9999.0+neuware113-cp39-cp39-manylinux2014_x86_64.whl
# 驗(yàn)證是否編譯成功
python3 -c "import megengine as mge; print(mge.is_cambricon_available()); print(mge.Tensor([1.]))"
- 輸出類似以下內(nèi)容，表示安裝成功
True
06 14:03:06[mgb] cambricon: card0: name=`MLU370' dyn_mem_reserve=0.00MiB alignment=0x100
Tensor([1.], device=xpux:0)
- MegEngine 寒武紀(jì)訓(xùn)練示例見 cambircon_train
關(guān)于寒武紀(jì)
寒武紀(jì)成立于 2016 年，專注于人工智能芯片產(chǎn)品的研發(fā)與技術(shù)創(chuàng)新，致力于打造人工智能領(lǐng)域的核心處理器芯片，讓機(jī)器更好地理解和服務(wù)人類。寒武紀(jì)提供云邊端一體、軟硬件協(xié)同、訓(xùn)練推理融合、具備統(tǒng)一生態(tài)的系列化智能芯片產(chǎn)品和平臺(tái)化基礎(chǔ)系統(tǒng)軟件。寒武紀(jì)產(chǎn)品廣泛應(yīng)用于服務(wù)器廠商和產(chǎn)業(yè)公司，面向互聯(lián)網(wǎng)、金融、交通、能源、電力和制造等領(lǐng)域的復(fù)雜 AI 應(yīng)用場(chǎng)景提供充裕算力，推動(dòng)人工智能賦能產(chǎn)業(yè)升級(jí)。
關(guān)于 MegEngine
開源深度學(xué)習(xí)框架曠視天元（MegEngine）是曠視自主研發(fā)的國(guó)產(chǎn)工業(yè)級(jí)深度學(xué)習(xí)框架 ，是曠視新一代 AI 生產(chǎn)力平臺(tái) Brain++ 的最核心組件，在2020年3月正式向全球開發(fā)者開源 。MegEngine 憑借其訓(xùn)練推理一體、超低硬件門檻和全平臺(tái)高效推理 3 大核心優(yōu)勢(shì)，能夠幫助企業(yè)與開發(fā)者大幅節(jié)省產(chǎn)品從實(shí)驗(yàn)室原型到工業(yè)部署的流程，真正實(shí)現(xiàn)小時(shí)級(jí)的轉(zhuǎn)化能力。
