Ubuntu16.04+cuda8.0+cudnn v6+tensorflow1.4

深度學習第一篇-----tensorflow環境搭建

   學習深度學習首先須要搭建開發環境,如今深度學習框架有不少,可是我仍是選擇google開源的tensorflow,安裝相對容易一點,適合初學者。因爲我是作研究用的,因此我下血本配置了一臺GTX1060-6g的顯卡。因爲個人電腦有顯卡,因此我選擇了裝支持gup版本的tensorflow。若是不須要裝支持gpu版本的能夠參考  http://blog.csdn.net/keith_bb/article/details/74066691這篇文章來安裝。因爲我買的帶有GPU的本子,我想不能浪費個人本子,因此我爬了無數坑裝好了支持gup版本的tensorflow。下面我就分享一下個人安裝過程。本次是用源碼來進行安裝的。html

環境框架:Ubuntu16.04+cuda8.0+cudnn v6+tensorflow1.4java

step1:python

下載所需的軟件包。linux

一、下載系統鏡像文件Ubuntu16.04  地址https://www.ubuntu.com/download/alternative-downloadsc++

                                                  

如圖所示下載鏡像文件。git

二、下載所需的cuda8.0(gtx1060也支持cuda8.0)github

地址https://developer.nvidia.com/cuda-downloadsweb

如今官網直接更新到9.0ubuntu

cuda 8.0地址https://developer.nvidia.com/cuda-80-ga2-download-archiveapi

三、下載cudnn v6

下載地址:https://developer.nvidia.com/cudnn

下載時可能須要登陸,註冊帳號後填寫調差問卷就能夠下載。

下載好所須要的安裝包就能夠進行按住安裝了。

step2:安裝

一、首先安裝顯卡GPU驅動

      打開終端:sudo apt-get update

      選擇系統設置→軟件更新→附加驅動→選擇nvidia最新驅動→應用更改.

二、安裝Tensorflow依賴的編譯工具bazel

      bazel安裝方法網址:https://bazel.build/versions/master/docs/install-ubuntu.html

(1) 安裝bazel前,需先安裝JDK8

      sudo apt-get installsoftware-properties-common

      sudo add-apt-repository ppa:webupd8team/java

      sudo apt-get update

      sudo apt-get installoracle-java8-installer

      驗證java版本:java -version

(2) 安裝bazel

      echo "deb [arch=amd64]http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee/etc/apt/sources.list.d/bazel.list

      sudo apt install curl

      curlhttps://bazel.build/bazel-release.pub.gpg | sudo apt-key add -

      sudo apt-get update

      sudo apt-get upgrade bazel

三、因爲cuda8.0不支持gcc 5.0以上的編譯器,所以須要降級,把編譯器版本降到4.9:

       先查看本身的gcc版本

        gcc –v

    下降gcc版本:

      sudoapt-get install g++-4.9

      sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20

      sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10

      sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20

      sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10

      sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30

      sudo update-alternatives --set cc /usr/bin/gcc

      sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30

      sudo update-alternatives --set c++ /usr/bin/g++

查看gcc版本

四、安裝cuda 8.0

切換目錄到cuda源文件下:

sudo dpkg -icuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb

sudo apt-get update

 sudo apt-get install cuda

五、安裝cuDNN 6.0

cp cudnn-8.0-linux-x64-v6.0.solitairetheme8 cudnn-8.0-linux-x64-v6.0.tgz

 tar -xvf cudnn-8.0-linux-x64-v6.0.tgz

 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*

六、配置環境變量

 sudo gedit ~/.bashrc

 export LD_LIBRARY_PATH=」$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64」

  export CUDA_HOME=/usr/local/cuda

  export PATH="$CUDA_HOME/bin:$PATH"

繼續在terminal中輸入:

source ~/.bash_profile #使更改的環境變量生效

七、安裝Tensflow

(1) 安裝Tensorflow依賴的其它工具包

      sudo apt-get install python-numpy swigpython-dev python-wheel

(2) 下載最新的Tensorflow源碼

      sudo apt-get install git

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

(3) 運行configure腳本配置環境信息(一路按回車建默認就行)或者參照下圖配置。

八、 經過bazel來編譯pip的安裝包,而後經過pip安裝

在terminal中輸入如下命令:(仍是在tensorflow目錄下)

Traceback (most recent call last):
  File "setup.py", line 25, in <module>
    from setuptools import find_packages, setup, Command
ImportError: No module named setuptools
setuptools-18.1https://www.cnblogs.com/huanbia/p/4725421.html
1.bazel build -c opt //tensorflow/tools/pip_package:build_pip_package
 2.bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
 3.bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg(編譯後的安裝包在這)
此處可能會出現以下問題:直安裝參考連接4.sudo pip install /home/***(你本身的用戶名)/Desktop/tensorflow-1.4.0-cp2-none-any.whl

上述第4部可能會出錯,須要安裝pip 包

命令:sudo apt-get install python-pip

以後執行第四部代碼效果以下:

如圖所示 安裝成功!!!!

相關文章
相關標籤/搜索
本站公眾號
   歡迎關注本站公眾號,獲取更多信息