近期要做一個人臉表情識別的課題,需要用到深度學習,鑒於指導老師推薦及未來應用前景的考慮,決定采用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.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /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