ubuntu14.04 + CUDA8 + cuDNN5 + Tensorflow


近期要做一個人臉表情識別的課題,需要用到深度學習,鑒於指導老師推薦及未來應用前景的考慮,決定采用tensorflow。

首先在自己的主機上安裝,然后在實驗室服務器(NVIDIA GTX TITANX)上安裝。 以下過程是在我的主機上安裝,TITANX上安裝方法是一樣的。

 

我的系統及環境:

    Ubuntu14.04(64位),  Python2.7, Nvidia GeForce GTX660

Tensorflow GPU 版本要求顯卡的Compute Capability >=  3.0, GTX660的Compute Capability =  3.0 (NVIDIA顯卡計算能力查看網址 https://developer.nvidia.com/cuda-gpus), 因此,我裝的是Tensorflow GPU版本。

安裝TF GPU版本,必要要先裝CUDA和cuDNN。

 

1.檢查環境

  lspci | grep -i nvidia                                        (查看顯卡型號)

  uname -m && cat /etc/*release

  gcc --version                                                     

  uname -r

  sudo apt-get install linux-headers-$(uname -r)                  (install與內核一致相關包)

 

2. CUDA 8安裝

  CUDA 8下載網址 https://developer.nvidia.com/cuda-downloads   如下圖,我下載的是這個版本。

 安裝過程:    

      sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb

      sudo apt-get update

      sudo apt-get install cuda

 安裝后CUDA后,需要設置環境變量, sudo gedit /etc/profile , 在profile結尾加上:

    export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}                (紅色字體為cuda的bin位置)

 保存后,在終端中 source /etc/profile

 

3. cuDNN 5安裝

    下載網址: https://developer.nvidia.com/cudnn   ,注冊后下載 cuDNN v5.1 Library for  Linux, 在終端中執行如下命令:

     tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz             ( 解壓cudnn)

     sudo cp -P cuda/include/cudnn./usr/local/cuda/include

     sudo cp -P cuda/lib64/libcudnn*  /usr/local/cuda/lib64

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

 

安裝完CUDA 8 和 cuDNN 5后, 在終端輸入 sudo apt-get install libcupti-dev

 

4. tensorflow安裝

   我選擇的安裝方式為 Install with native pip (https://www.tensorflow.org/install/install_linux)

安裝pip:   sudo apt-get install python-pip python-dev

下載tensorflow(GPU版):https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl 

 安裝tensorflow:

      (1)在線安裝 

                sudo pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl 

   (2)下載安裝(由於Ubuntu系統下,網上比較慢,可以在windows下載。推薦這種安裝方法) 

               sudo pip install tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl               

 

5. 測試tensorflow

終端輸入:  python

>>> import tensorflow as tf

  

>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

 

 

參考網址 

https://github.com/tensorflow/tensorflow

https://www.tensorflow.org/install/install_linux

http://www.qingpingshan.com/pc/fwq/172247.html


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM