Centos 7.4 安裝GPU版tensorflow

安裝nvidia驅動:

CentOS 7.0 Nvidia顯卡安裝步驟:

1 在英偉達官網下載相應驅動:

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

2 屏蔽默認帶有的nouveau:

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

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

將nvidiafb注釋掉。

#blacklist nvidiafb

然后添加以下語句:

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 修改運行級別為文本模式:

# systemctl set-default multi-user.target

5 重新啟動, 使用root用戶登陸:

# reboot

6 進入下載的驅動所在目錄:

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

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

安裝過程中,選擇accep,其余默認選擇yes或者install

7 修改運行級別回圖形模式:

# systemctl set-default graphical.target

8 重新啟動:

# reboot

9 驗證,結果如下圖則成功:

$ 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配置環境變量:

$ vim ~/.bashrc

添加以下內容:

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執行以下指令后重啟:

source ~/.bashrc?

5.驗證,結果顯示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.? ? 若提示以下錯誤:

“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.“

執行以下指令安裝kernel-devel包:

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

2.? ? 報錯“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網站下載安裝包,需要注冊,(這里選用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官網下載安裝包:

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

2?用rpm安裝:

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

3?配置環境變量(具體jdk版本號需要進入/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?進入解壓后的文件夾進行配置:

$ cd tensorflow

$ ./configure

參考配置如下:



4?編譯(gpu版),這個過程有點長:

$ 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環境下執行“import tensorflow”,如無報錯,則安裝成功。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,247評論 6 543
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,520評論 3 429
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 178,362評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,805評論 1 317
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,541評論 6 412
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,896評論 1 328
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,887評論 3 447
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 43,062評論 0 290
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,608評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,356評論 3 358
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,555評論 1 374
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,077評論 5 364
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,769評論 3 349
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,175評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,489評論 1 295
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,289評論 3 400
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,516評論 2 379