Centos 7.4 安裝GPU版tensorflow

安裝nvidia驅(qū)動(dòng):

CentOS 7.0 Nvidia顯卡安裝步驟:

1 在英偉達(dá)官網(wǎng)下載相應(yīng)驅(qū)動(dòng):

http://www.nvidia.com/download/driverResults.aspx/122825/en-us

2 屏蔽默認(rèn)帶有的nouveau:

使用su命令切換到root用戶下: su root

打開/lib/modprobe.d/dist-blacklist.conf

將nvidiafb注釋掉。

#blacklist nvidiafb

然后添加以下語(yǔ)句:

blacklist nouveau

options nouveau modeset=0

3 重建initramfs image步驟:

# mv /boot/initramfs-(uname?r).img /boot/initramfs?(uname -r).img.bak

# dracut /boot/initramfs-(uname?r).img(uname?r)

4 修改運(yùn)行級(jí)別為文本模式:

# systemctl set-default multi-user.target

5 重新啟動(dòng), 使用root用戶登陸:

# reboot

6 進(jìn)入下載的驅(qū)動(dòng)所在目錄:

# chmod +x NVIDIA-Linux-x86_64-346.47.run

# ./NVIDIA-Linux-x86_64-346.47.run

安裝過程中狰闪,選擇accep疯搅,其余默認(rèn)選擇yes或者install

7 修改運(yùn)行級(jí)別回圖形模式:

# systemctl set-default graphical.target

8 重新啟動(dòng):

# reboot

9 驗(yàn)證,結(jié)果如下圖則成功:

$ nvidia-smi


安裝cuda:

1下載安裝包:

https://developer.nvidia.com/cuda-80-ga2-download-archive

選擇rpm(local)埋泵,如下:


2安裝指令如下:

$ sudo rpm -i cuda-repo-rhel7-8-0-local-ga2-8.0.61-1.x86_64.rpm

$ sudo yum clean all

$ sudo yum install cuda


3配置環(huán)境變量:

$ vim ~/.bashrc

添加以下內(nèi)容:

export CUDA_HOME=/usr/local/cuda-8.0?

export PATH=/usr/local/cuda-8.0/bin:$PATH?

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH?

export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib:${LD_LIBRARY_PATH}"

4執(zhí)行以下指令后重啟:

source ~/.bashrc?

5.驗(yàn)證幔欧,結(jié)果顯示PASS則安裝成功:

$ cd /usr/local/cuda-8.0/samples/1_Utilities/bandwidthTest/

$ sudo make

$ cd /usr/local/cuda-8.0/samples/

$ ./bin/x86_64/linux/release/bandwidthTest


備注:

1.? ? 若提示以下錯(cuò)誤:

“error:unable to find the kernel source tree for the currently running kernel. please make sure you have installed the kernel source files for your kernel and that they are properly configured; on red hat linux system, for example, be sure you have the 'kernel-source' or 'kernel-devel' RPM installed. if you know the correct kernel source files are installed ,you may specify the kernel source path with the '--kernel-source-path' command line option.“

執(zhí)行以下指令安裝kernel-devel包:

$ sudo yum install kernel-devel-$(uname -r)

2.? ? 報(bào)錯(cuò)“Error: Package:1:nvidia-kmod-375.51-2.el7.x86_64 (cuda) Requires: dkms“

安裝dkms依賴:

$ sudo yum install epel-release

$ sudo yum install dkms


安裝cuDNN:

1.???nvidia網(wǎng)站下載安裝包,需要注冊(cè)丽声,(這里選用cudnn-8.0-linux-x64-v6.0版礁蔗,下載的后綴名可能不一樣,直接改成.tgz就可以了):

https://developer.nvidia.com/cudnn

2.????解壓生成cuda文件夾:

$ tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz

3.???拷貝文件到CUDA安裝目錄:

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include

$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安裝jdk:

1?在oracle官網(wǎng)下載安裝包:

http://www.oracle.com/technetwork/java/javase/downloads/jdk8-downloads-2133151.html

2?用rpm安裝:

$ sudo rpm -i jdk-8u91-linux-x64.rpm

3?配置環(huán)境變量(具體jdk版本號(hào)需要進(jìn)入/usr/java文件夾查看):

$ export JAVA_HOME=/usr/java/jdk1.8.0_91


安裝bazel:

1?下載指定版本的安裝腳本:

https://github.com/bazelbuild/bazel/releases

2?修改文件屬性:

$ chmod +x bazel--installer-linux-x86_64.sh

3?安裝:

$./bazel--installer-linux-x86_64.sh --user

4?配置PATH變量:

$ vim ~/.bashrc

添加:export PATH="$PATH:$HOME/bin"


安裝tensorflow:

1?安裝依賴:

$ sudo yum -y install numpy swig python-devel python-wheel python-pip?zlib zlib-devel

2?在github上下載安裝文件.zip雁社,并解壓:

https://github.com/tensorflow/tensorflow

$ unzip tensorflow*.zip

3?進(jìn)入解壓后的文件夾進(jìn)行配置:

$ cd tensorflow

$ ./configure

參考配置如下:



4?編譯(gpu版)浴井,這個(gè)過程有點(diǎn)長(zhǎng):

$ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

5 生成.whl包:

$ bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

6?安裝生成的包:

$ sudo pip install /tmp/tensorflow_pkg/tensorflow-*.whl

7?重啟。

8?命令行輸入“python“霉撵,在python環(huán)境下執(zhí)行“import tensorflow”偏化,如無(wú)報(bào)錯(cuò)励翼,則安裝成功闲先。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末骡和,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子崭参,更是在濱河造成了極大的恐慌,老刑警劉巖款咖,帶你破解...
    沈念sama閱讀 218,386評(píng)論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件何暮,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡铐殃,警方通過查閱死者的電腦和手機(jī)海洼,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,142評(píng)論 3 394
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)富腊,“玉大人坏逢,你說我怎么就攤上這事∽副唬” “怎么了是整?”我有些...
    開封第一講書人閱讀 164,704評(píng)論 0 353
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)民假。 經(jīng)常有香客問我浮入,道長(zhǎng),這世上最難降的妖魔是什么羊异? 我笑而不...
    開封第一講書人閱讀 58,702評(píng)論 1 294
  • 正文 為了忘掉前任事秀,我火速辦了婚禮彤断,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘易迹。我一直安慰自己宰衙,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,716評(píng)論 6 392
  • 文/花漫 我一把揭開白布睹欲。 她就那樣靜靜地躺著供炼,像睡著了一般。 火紅的嫁衣襯著肌膚如雪句伶。 梳的紋絲不亂的頭發(fā)上劲蜻,一...
    開封第一講書人閱讀 51,573評(píng)論 1 305
  • 那天,我揣著相機(jī)與錄音考余,去河邊找鬼先嬉。 笑死,一個(gè)胖子當(dāng)著我的面吹牛楚堤,可吹牛的內(nèi)容都是我干的疫蔓。 我是一名探鬼主播,決...
    沈念sama閱讀 40,314評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼身冬,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼衅胀!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起酥筝,我...
    開封第一講書人閱讀 39,230評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤滚躯,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后嘿歌,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體掸掏,經(jīng)...
    沈念sama閱讀 45,680評(píng)論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,873評(píng)論 3 336
  • 正文 我和宋清朗相戀三年宙帝,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了丧凤。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,991評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡步脓,死狀恐怖愿待,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情靴患,我是刑警寧澤仍侥,帶...
    沈念sama閱讀 35,706評(píng)論 5 346
  • 正文 年R本政府宣布,位于F島的核電站鸳君,受9級(jí)特大地震影響访圃,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜相嵌,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,329評(píng)論 3 330
  • 文/蒙蒙 一腿时、第九天 我趴在偏房一處隱蔽的房頂上張望况脆。 院中可真熱鬧,春花似錦批糟、人聲如沸格了。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,910評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)盛末。三九已至,卻和暖如春否淤,著一層夾襖步出監(jiān)牢的瞬間悄但,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,038評(píng)論 1 270
  • 我被黑心中介騙來(lái)泰國(guó)打工石抡, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留檐嚣,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,158評(píng)論 3 370
  • 正文 我出身青樓啰扛,卻偏偏與公主長(zhǎng)得像嚎京,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子隐解,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,941評(píng)論 2 355