tensorflow for GPU build from source

深度學習主機環(huán)境配置:

Ubuntu16.04+Nvidia GTX 1080Ti+CUDA9.0+cuDNN7.0.3+nvidia driver 384

1.安裝Ubuntu16.04,安裝完畢后Ubuntu

16.04的分辨率很低.

更新:

sudo apt-get upgrade

2.查看gcc版本號

gcc-v

目前的cuda8.0已支持gcc

5.4版本了谦纱。

2.安裝GTX1080驅動

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get update

sudo apt-get install nvidia-367

sudo apt-getinstall mesa-common-dev

sudo apt-get install freeglut3-dev

以上方法是目前最為穩(wěn)定的簡單的顯卡驅動安裝方法御毅,之后重啟系統(tǒng)讓GTX1080顯卡驅動生效和蚪。

3.下載和安裝CUDA

下載的“cuda_8.0.61_linux.run”有1.45G,按照Nivdia官方給出的方法安裝CUDA8:

sudo sh cuda_8.0.27_linux.run --tmpdir=/opt/temp/

提示中:Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 361.62?

答案必須是n姥敛,否則之前安裝的GTX1080驅動就白費了抓半,而且問題多多姚炕。

安裝完畢后,再聲明一下環(huán)境變量捌显,并將其寫入到~/.bashrc的尾部:

在home處按下ctrl+h顯示隱藏文件,然后編輯bashrc文件总寒,添加一下兩行到最后扶歪。

export PATH=/usr/local/cuda-8.0/bin\${PATH:+:\${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64\${LD_LIBRARY_PATH:+:\${LD_LIBRARY_PATH}}

然后source ~/.bashrc更新文件。

最后再來測試一下CUDA摄闸,運行:

nvidia-smi

nvcc-V

因為我們不是從官網(wǎng)的run文件安裝的驅動善镰,我們要手動將將驅動路徑加入到環(huán)境變量LIBRARY_PATH中,I

have nvidia-367 driver and I didn't install the driver from the run

file. Would you please help me how I can fix this?

Analternative is to set LIBRARY_PATH to include /use/lib/nvidia-*/like:

$ LIBRARY_PATH=/usr/lib/nvidia-你的版本號:$LIBRARY_PATH make

測試cuda8.0是否安裝成功,NVIDIA_CUDA-8.0_Samples安裝目錄下年枕,運行以上的make命令炫欺。

大約十分鐘后,顯示bulidfinished熏兄。

$cd ./NVIDIA_CUDA-8.0_Samples/bin/x86_64/linux/release/

$./deviceQuery

提示如下錯誤:

zyl@zyl-PC:/usr/local/cuda/samples/1_Utilities/deviceQuery$ ./deviceQuery./deviceQuery Starting...CUDA Device Query (Runtime API) version (CUDART static linking)modprobe: ERROR: could not insert 'nvidia_340_uvm': Invalid argumentcudaGetDeviceCount returned 30-> unknown errorResult = FAIL

嘗試解決:sudo reboot

安裝cudnn5.1

csdn上下載品洛,CSDN提供了下載地址:http://download.csdn.net/download/hearthougan/9889201。注冊一個帳號送3分摩桶,免費下載資源桥状。下載Cudnnv5.1,進入下載目錄硝清,執(zhí)行下列命令:

tar xvzf cudnn-8.0-Linux-x64-v5.1.tgz

解壓完成辅斟。

添加頭文件和庫文件

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

sudo cp lib64/libcudnn.* /usr/local/lib

添加系統(tǒng)環(huán)境變量:/etc/profile文件中,

$sudo gedit /etc/profile,文件末尾加上一行保存:export PATH = /usr/local/cuda/bin:$PATH

創(chuàng)建鏈接文件:sudo gedit /etc/ld.so.conf.d/cuda.conf,文件中加上一行保存:/usr/local/cuda/lib64

$sudo gedit /etc/profile

在最后添加export PATH = /usr/local/cuda/bin:$PATH

$cd /usr/local/cuda/lib64/

$ rm -rf libcudnn.so libcudnn.so.5? ? #刪除原有動態(tài)文件sudo mv $sudo/usr/lib/nvidia-375/libEGL.so.1 /usr/lib/nvidia-375/libEGL.so.1.org

$sudo mv /usr/lib32/nvidia-375/libEGL.so.1 /usr/lib32/nvidia-375/libEGL.so.1.org

$sudo ln -s /usr/lib/nvidia-375/libEGL.so.375.39 /usr/lib/nvidia-375/libEGL.so.1

$sudo ln -s /usr/lib32/nvidia-375/libEGL.so.375.39 /usr/lib32/nvidia-375/libEGL.so.1

sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5? #生成軟銜接

sudo ln -s libcudnn.so.5 libcudnn.so? ? ? #生成軟鏈接

安裝tensorflow

安裝bazel

由于本教程使用tensorflow源碼編譯/安裝芦拿,所以需要使用bazel

build士飒。鏈接:https://www.bazel.io/versions/master/docs/install.html


安裝第三方庫

$ sudo apt-get install python-numpy swig python-dev python-wheel

$sudo apt-get install gitgit clone git://github.com/numpy/numpy.git numpy

下載tensorflow

在terminal中輸入以下命令

git clone https://github.com/tensorflow/tensorflow

默認下載目錄是在/home下

在terminal中輸入以下命令:

cd ~/tensorflow #切換到tensorflow文件夾

./configure #執(zhí)行configure文件

bazel build -c opt //tensorflow/tools/pip_package:build_pip_package

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

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

cd /tmp/tensorflow_pkg

ls#顯示了生成的wheel

sudo pip install 上面生成的wheel文件


設置環(huán)境

bazel build -c opt //tensorflow/tools/pip_package:build_pip_package# To build with GPU support:

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

mkdir _python_build

cd _python_build

ln -s ../bazel-bin/tensorflow/tools/pip_package/build_pip_package.runfiles/org_tensorflow/* .

ln -s ../tensorflow/tools/pip_package/* .python setup.py develop

最后建議$

sudo apt-get install python-pip

用pip安裝jupyter

notebook以及其它的安裝包

完美運行查邢,可復制。

以上僅作參考变汪。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末侠坎,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子裙盾,更是在濱河造成了極大的恐慌实胸,老刑警劉巖,帶你破解...
    沈念sama閱讀 211,290評論 6 491
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件番官,死亡現(xiàn)場離奇詭異庐完,居然都是意外死亡,警方通過查閱死者的電腦和手機徘熔,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,107評論 2 385
  • 文/潘曉璐 我一進店門门躯,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人酷师,你說我怎么就攤上這事讶凉。” “怎么了山孔?”我有些...
    開封第一講書人閱讀 156,872評論 0 347
  • 文/不壞的土叔 我叫張陵懂讯,是天一觀的道長。 經(jīng)常有香客問我台颠,道長褐望,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,415評論 1 283
  • 正文 為了忘掉前任串前,我火速辦了婚禮瘫里,結果婚禮上,老公的妹妹穿的比我還像新娘荡碾。我一直安慰自己谨读,他們只是感情好,可當我...
    茶點故事閱讀 65,453評論 6 385
  • 文/花漫 我一把揭開白布坛吁。 她就那樣靜靜地躺著漆腌,像睡著了一般。 火紅的嫁衣襯著肌膚如雪阶冈。 梳的紋絲不亂的頭發(fā)上闷尿,一...
    開封第一講書人閱讀 49,784評論 1 290
  • 那天,我揣著相機與錄音女坑,去河邊找鬼填具。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的劳景。 我是一名探鬼主播誉简,決...
    沈念sama閱讀 38,927評論 3 406
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼盟广!你這毒婦竟也來了闷串?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 37,691評論 0 266
  • 序言:老撾萬榮一對情侶失蹤筋量,失蹤者是張志新(化名)和其女友劉穎烹吵,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體桨武,經(jīng)...
    沈念sama閱讀 44,137評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡肋拔,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,472評論 2 326
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了呀酸。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片凉蜂。...
    茶點故事閱讀 38,622評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖性誉,靈堂內的尸體忽然破棺而出窿吩,到底是詐尸還是另有隱情,我是刑警寧澤错览,帶...
    沈念sama閱讀 34,289評論 4 329
  • 正文 年R本政府宣布爆存,位于F島的核電站,受9級特大地震影響蝗砾,放射性物質發(fā)生泄漏。R本人自食惡果不足惜携冤,卻給世界環(huán)境...
    茶點故事閱讀 39,887評論 3 312
  • 文/蒙蒙 一悼粮、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧曾棕,春花似錦扣猫、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,741評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至衙耕,卻和暖如春昧穿,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背橙喘。 一陣腳步聲響...
    開封第一講書人閱讀 31,977評論 1 265
  • 我被黑心中介騙來泰國打工时鸵, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 46,316評論 2 360
  • 正文 我出身青樓饰潜,卻偏偏與公主長得像初坠,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子彭雾,可洞房花燭夜當晚...
    茶點故事閱讀 43,490評論 2 348

推薦閱讀更多精彩內容