TensorFlow通過AMD GPU加速(ROCm/elementary OS 5.0)

elementary OS 5.0是基于Ubuntu 18.04制作尤泽,所以本流程也適用于Ubuntu 18.04欣簇。

ROCm升級1.91之后不需要安裝AMD的GPU驅(qū)動程序规脸,如果安裝可能會出現(xiàn)問題。

安裝elementary OS 5 之后的一些配置

詳細配置請參閱:elementary OS 5.0安裝后配置

安裝軟件更新管理器

sudo apt update
sudo apt install software-properties-common software-properties-gtk

運行軟件更新管理器然后選擇合適的鏡像醉蚁,我使用的是阿里云燃辖。

sudo software-properties-gtk

更新系統(tǒng)

sudo apt update
sudo apt upgrade

安裝ROCm

添加ROCm庫

wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -
sudo sh -c 'echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main > /etc/apt/sources.list.d/rocm.list'

安裝必要的軟件包

sudo apt update && sudo apt install -y \
  build-essential \
  clang \
  clang-format \
  clang-tidy \
  cmake \
  cmake-qt-gui \
  ssh \
  curl \
  apt-utils \
  pkg-config \
  g++-multilib \
  git \
  libunwind-dev \
  libfftw3-dev \
  libelf-dev \
  libncurses5-dev \
  libpthread-stubs0-dev \
  vim \
  gfortran \
  libboost-program-options-dev \
  libssl-dev \
  libboost-dev \
  libboost-system-dev \
  libboost-filesystem-dev \
  rpm \
  wget 

安裝ROCm驅(qū)動

sudo apt update && \
    sudo apt install -y --allow-unauthenticated \
    rocm-dkms rocm-dev rocm-libs \
    rocm-device-libs \
    hsa-ext-rocr-dev hsakmt-roct-dev hsa-rocr-dev \
    rocm-opencl rocm-opencl-dev \
    rocm-utils \
    rocm-profiler cxlactivitylogger \
    miopen-hip miopengemm

添加用戶

sudo adduser $LOGNAME video
sudo reboot

安裝miniconda

在使用python時我們很多時候會用到conda鬼店,但是conda會改變python的一些配置网棍。所以我們先安裝conda,這樣就不會出現(xiàn)在使用tensorflow時需要使用/usr/bin/python3這樣的命令妇智。
下載地址(64位):https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh

cd 下載目錄
chmod +x ./Miniconda3-latest-Linux-x86_64.sh
./Miniconda3-latest-Linux-x86_64.sh

安裝之后我們發(fā)現(xiàn)conda的python是3.7滥玷,而ROCm版的tensorflow支持3.5和3.6,所以我們需要將python降為3.6巍棱。

conda install python=3.6

安裝python相關(guān)的軟件包

sudo apt update && sudo apt install -y \
    python3-numpy \
    python3-dev \
    python3-wheel \
    python3-mock \
    python3-future \
    python3-pip \
    python3-yaml \
    python3-setuptools

安裝TensorFlow

pip3 install --user tensorflow-rocm
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末惑畴,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子航徙,更是在濱河造成了極大的恐慌如贷,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,451評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件到踏,死亡現(xiàn)場離奇詭異杠袱,居然都是意外死亡,警方通過查閱死者的電腦和手機窝稿,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,172評論 3 394
  • 文/潘曉璐 我一進店門楣富,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人伴榔,你說我怎么就攤上這事纹蝴。” “怎么了踪少?”我有些...
    開封第一講書人閱讀 164,782評論 0 354
  • 文/不壞的土叔 我叫張陵塘安,是天一觀的道長。 經(jīng)常有香客問我援奢,道長兼犯,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,709評論 1 294
  • 正文 為了忘掉前任萝究,我火速辦了婚禮免都,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘帆竹。我一直安慰自己绕娘,他們只是感情好,可當我...
    茶點故事閱讀 67,733評論 6 392
  • 文/花漫 我一把揭開白布栽连。 她就那樣靜靜地躺著险领,像睡著了一般侨舆。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上绢陌,一...
    開封第一講書人閱讀 51,578評論 1 305
  • 那天挨下,我揣著相機與錄音,去河邊找鬼脐湾。 笑死臭笆,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的秤掌。 我是一名探鬼主播愁铺,決...
    沈念sama閱讀 40,320評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼闻鉴!你這毒婦竟也來了茵乱?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,241評論 0 276
  • 序言:老撾萬榮一對情侶失蹤孟岛,失蹤者是張志新(化名)和其女友劉穎瓶竭,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體渠羞,經(jīng)...
    沈念sama閱讀 45,686評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡斤贰,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,878評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了堵未。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片腋舌。...
    茶點故事閱讀 39,992評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖渗蟹,靈堂內(nèi)的尸體忽然破棺而出块饺,到底是詐尸還是另有隱情,我是刑警寧澤雌芽,帶...
    沈念sama閱讀 35,715評論 5 346
  • 正文 年R本政府宣布授艰,位于F島的核電站,受9級特大地震影響世落,放射性物質(zhì)發(fā)生泄漏淮腾。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,336評論 3 330
  • 文/蒙蒙 一屉佳、第九天 我趴在偏房一處隱蔽的房頂上張望谷朝。 院中可真熱鬧,春花似錦武花、人聲如沸圆凰。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,912評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽专钉。三九已至挑童,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間跃须,已是汗流浹背站叼。 一陣腳步聲響...
    開封第一講書人閱讀 33,040評論 1 270
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留菇民,地道東北人尽楔。 一個月前我還...
    沈念sama閱讀 48,173評論 3 370
  • 正文 我出身青樓,卻偏偏與公主長得像玉雾,于是被迫代替她去往敵國和親翔试。 傳聞我的和親對象是個殘疾皇子轻要,可洞房花燭夜當晚...
    茶點故事閱讀 44,947評論 2 355

推薦閱讀更多精彩內(nèi)容