您好,歡迎來電子發(fā)燒友網(wǎng)! ,新用戶?[免費注冊]

您的位置:電子發(fā)燒友網(wǎng)>源碼下載>數(shù)值算法/人工智能>

AMD開源的高性能機器學習庫MIOpen分析

大?。?/span>0.5 MB 人氣: 2017-09-28 需要積分:1

  MIOpen是AMD開源的高性能機器學習庫,目前只支持兩種編程模式:

  1. OpenCL

  2. HIP

  安裝準備

  安裝ROCm

  基礎軟件棧,其中包括

  OpenCL - OpenCL 庫和頭文件

  HIP -

  HIP 和 HCC庫和頭文件

  clang-ocl– 必需

  MIOpen依靠miopengemm庫來啟用

  ROCm cmake模塊可以從這里安裝:https://github.com/RadeonOpenCompute/rocm-cmake

  對應的安裝說明可在下載的庫中查詢。

  cmake配置

  第一步:創(chuàng)建目錄:

  第二步:配置cmake。MIOpen 的首選后端可以使用-DMIOPEN_BACKEND cmake 變量來設置。

  For OpenCL, run:

  以上步驟均假設 OpenCL 安裝在某個標準位置。如果沒有,則手動設置這兩個 cmake 變量:

  For HIP, run:

  cmake 步驟示例:

  默認安裝位置設置為 “/opt/rocm”, 可使用 CMAKE_INSTALL_PREFIX設置:

  此外, 網(wǎng)絡配置的數(shù)據(jù)庫路徑可由 MIOPEN_DB_PATH 變量設置。默認為數(shù)據(jù)庫文件的安裝位置。出于開發(fā)便捷性, 設置 BUILD_DEV 將改變存儲在源目錄中的數(shù)據(jù)庫文件的路徑:

  使用 ccmake 運行 cmake 后,可以更改配置:

  ccmake 程序在 不兼容windows平臺。

  構建庫

  可以使用“Release”配置在build目錄構建庫。

  用 “install”命令進行目標安裝:

  此時的安裝路徑為已設置的CMAKE_INSTALL_PREFIX。

  驅動生成

  MIOpen 提供了一個應用驅動程序,可用于單獨執(zhí)行任何一個特定的層,并測量和驗證庫的性能。

  可以使用 MIOpenDriver 目標構建驅動程序:

  有關如何運行驅動程序的文檔在這里:https://github.com/ROCmSoftwarePlatform/MIOpen/blob/master/driver/README.md

  運行測試

  可以使用 target check 運行測試:

  通過執(zhí)行以下操作, 可以生成和運行單個測試:

  生成文檔

  HTML和PDF文檔生成命令:

  這將在。/MIOpen/doc/html 文件夾建立一個本地可搜索網(wǎng)站, 并在。/MIOpen/doc/pdf 文件夾創(chuàng)建一個 PDF 文檔。

  文檔是使用Doxygen生成的, 應單獨安裝。

  HTML和PDF通過Sphinx和Breathe以及ReadTheDocs theme生成。

  可以在MIOpen / doc文件夾中填寫Sphinx,Breathe和ReadTheDocs theme的要求:

  由于設置不同,有可能需要執(zhí)行sudo才能安裝pip 。

  代碼格式化

  代碼格式化使用 clang-format。格式化一個文件,使用:

  此外,可以安裝githook來格式化per-commit代碼

非常好我支持^.^

(0) 0%

不好我反對

(0) 0%

      發(fā)表評論

      用戶評論
      評價:好評中評差評

      發(fā)表評論,獲取積分! 請遵守相關規(guī)定!

      ?