在Ubuntu18 04上安裝tvm

2021-10-08 18:43:01 字數 2504 閱讀 3258

工欲善其事，必先利其器。

在開始學習和使用tvm之前，必定是要先安裝乙個的…

不過直至2023年8月，tvm發布的最新版本還是0.* 的，尚未發布正式版。安裝上也有許多坑要填，不過按照官方文件上操作也基本是穩的。

本文接下來的內容將簡要介紹一下筆者安裝tvm的過程。

平台及版本

前排建議：如果不喜歡折騰，盡量別在windows上安裝，會有很多莫名其妙的坑要填，而且未必有比較好的解決方案（親測在win10上安裝失敗了5555）

硬體：x86伺服器 + gtx1080ti gpu

系統：ubuntu18.04

tvm版本：0.7.dev1

llvm版本：10.0.0

安裝過程

ubuntu上安裝tvm只要根據官網教程說明操作就基本沒啥問題了。官方提供了從原始碼安裝和docker映象兩種安裝方法，這裡介紹從原始碼安裝的方法。

先cd到你想存放tvm的目錄下

再終端執行如下指令：

git clone --recursive  tvm
記得需要加上–recursive

step2：安裝依賴包

執行以下兩條語句即可，實際上大多數包應該都已經在之前安裝過了，沒有啥特殊的依賴。

sudo apt-get update


sudo apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
如果需要提供cuda支援，還需要先安裝好cuda（最低版本需求是8.0），這個就不做具體安裝方法的介紹了，網上一搜一堆，有nvidia顯示卡的還是盡量安乙個吧。

官網還強烈推薦安裝llvm，這是乙個能夠優化以任何語言編寫的程式的編譯效率的工具，tvm本身也是乙個深度學習編譯器，因此需要llvm幫助提公升效率。但llvm的編譯安裝也是很讓人頭大的事情，接下來介紹如何快速有效地安裝llvm。

step3：安裝llvm（和clang）

接下來需要新增一些ubuntu的環境變數，在/etc/profile/檔案中新增以下兩行：

export llvm_home=你剛解壓的llvm資料夾的絕對路徑/bin


export path=$llvm_home:$path
step4：配置並編譯tvm
mkdir build


cp cmake/config.cmake build
然後進行編譯配置，修改build/config.cmake檔案內容：

接下來就可以進行編譯了：

cd build


cmake ..


make -j4	# 也可以設為-j8，提公升編譯速度
如果出現類似如下的提示，則說明編譯成功了

安裝tvm

官網提供了兩種tvm的安裝方法，一種是設定系統環境變數，這種適合打算修改tvm原始碼的開發者，另一種是用提供的python指令碼安裝，這裡介紹使用python指令碼的安裝。

分別用python直譯器執行tvm目錄下python資料夾內的setup.py，以及topi/python目錄下的setup.py，具體指令如下：

cd python	# 這裡指的是git clone得到的tvm資料夾下的python資料夾


python setup.py install --user	# 如果打算用python虛擬環境，得先把虛擬環境開了再執行這條


cd ..


cd topi/python


python setup.py install --user
再用pip安裝一些依賴的python包：

pip3 install --user numpy decorator attrs		# 必須安裝


pip3 install --user tornado psutil xgboost	# 可選


sudo apt install antlr4									# 可選


pip3 install --user mypy orderedset antlr4-python3-runtime	# 可選
如果你的python環境是虛擬環境，則不需要加上–user

測試tvm是否安裝成功

開啟你的python直譯器，執行如下指令：

>>> import tvm


>>> tvm.__version__
如果匯入tvm時候沒有報錯，並且正確輸出了tvm的版本，那麼恭喜你，安裝成功！

在Ubuntu18 04上安裝anaconda3

清華大學開源軟體映象站 安裝包從這裡下在 直接拉到最下邊，選擇最新版 或者你需要的版本 在命令列下輸入 wget anaconda3 5.3.1 linux x86 64.shsudo sh anaconda3 5.3.1 linux x86 64.sh之後按enter 在經歷的一堆more之後，問...

在Ubuntu18 04上安裝 Wine 4 0

2019.07.13 13 21 55 字數 453 閱讀 129 在本節中，我們將提供有關如何在 ubuntu 18.04 上安裝 wine 4.0 的分步說明。首先，使用以下 wget 命令將 winehq 儲存庫的 gpg 金鑰匯入您的系統 wget qo sudo apt key add 通...

gsoap在ubuntu18 04上編譯

如果從來沒安裝過，預先安裝以下內容。apt get install g flex bison openssl libssl dev libgtk2.0 dev libglib2.0 dev 之後進入源 檔案件 gsoap 2.8 下執行如下命令 configure prefix usr local ...