Ubuntu 安裝 tensorflow-gpu + keras


轉載請注明:

仰望大牛的小清新http://www.cnblogs.com/luruiyuan/

本文原網址http://www.cnblogs.com/luruiyuan/p/6660142.html

我用的Ubuntu版本是 16.04 ,並且使用了 gnome 作為桌面(這一點無關緊要),經歷了許多波折,終於完成了以 tensorflow 為后端的 keras 的安裝。

tensorflow-GPU 版本的安裝:

1.下載 CUDA 8.0

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

安裝如下版本:

2.下載 cuDNN v5 (這里我下載的是 v6, 但是事實證明 tensorflow 不直接支持 v6,除非你自己編譯時制定了 v6,否則按照如下教程安裝的 tensorflow 只能加載 v5)

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

這里需要先 登錄/注冊 后才能下載

3.安裝 NVIDIA 驅動:

首先打開 terminal,輸入一下指令,更新應用列表

sudo apt-get update

多虧了 Linux 社區的強大支持,我們可以用很簡潔的方式安裝驅動:

按下 win 按鍵,打開菜單,如圖:

然后在上方的 type to search 中輸入: additional drivers 打開 "additional drivers —— 附加驅動",然后選擇 與自己顯卡匹配的 NVIDIA 驅動,我這里是 

using nvidia binary drivers (375)

然后選擇 應用更改,這里安裝可能會失敗,此時你可以繼續選擇應用更改,多安裝幾次即可

安裝完成后,選擇 restart

4. gcc 降版本

CUDA8.0 不支持 gcc 5.0 及以上的編譯器,而系統自帶的是5.4及以上版本,因此我們需要降版本,否則會在后面報錯

在terminal中輸入如下指令,將 gcc 版本降到 4.9

sudo apt-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++ 

5.安裝 CUDA 8.0

cd /media/你的用戶名/TOSHIBA\ EXT/alu/CUDA/ # 進入cuda 所在文件夾
# 根據官網上的提示安裝 cuda 8.0
sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64​.deb
sudo apt-get update
sudo apt-get install cuda​

6. 安裝 cuDNN

cd  cd /media/你的用戶名/TOSHIBA\ EXT/alu/CUDA/ # 進入 cuDNN 安裝文件的所在路徑
tar xvzf cudnn-8.0-linux-x64-v6.0.tgz # 解壓
sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 復制到 include 中
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 # 復制到 lib64 中
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 講頭文件復制進去

7. 配置CUDA環境變量

開啟 GPU 支持:

根據官網教程

我們在terminal中鍵入下列命令:

sudo gedit ~/.bash_profile # 打開.bash_profile 這是用戶的環境變量,不是全局的

然后在打開的文本末尾加入:

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

保存並關閉后,輸入下列命令使環境變量生效:

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

安裝完成后,一定要通過如下兩條命令對驅動進行檢驗:

1. NVIDIA的設置界面

nvidia-settings # 打開 NVIDIA 設置界面

這條指令打開的界面如下:

2. NVIDIA GPU 列表

nvidia-smi

這條指令會在 terminal 中產生GPU列表,如,我這里只有一個GPU

網上有些人抄別人的博客,說 nvcc -V 就可以驗證,經過我實測,存在 nvcc -V 正常輸出但是驅動仍安裝失敗的現象,因此,上述驗證方法是不可信的

8.安裝 python3.5.2

由於tensorflow1.0 對於 python3 支持更好,並且目前只支持python3.5.2,因此我們選擇 python 3.5.2。

使用 Anaconda3-4.2.0-Linux-x86_64.sh 進行安裝,地址如下:

https://repo.continuum.io/archive/

安裝完成后,添加環境變量,將其設置為默認的 python 解釋器

首先打開環境變量的文件

gedit ~/.bashrc

然后在文件末尾加入 anaconda3 的路徑

export PATH=/home/你的路徑/anaconda3/bin:$PATH

最后使我們的改動生效

source ~/.bashrc

這樣,我們在terminal中輸入 python 就會默認打開 anaconda3

這樣我們就可以安心使用python3了。

9.安裝 keras 和 tensorflow

有了上述安裝過程,我們系統中默認的pip將會是anaconda3中自帶的pip,這樣我們只需要使用pip即可安裝 keras 和 tensorflow 到 anaconda 中。

執行如下命令:

pip install tensorflow-gpu keras # 安裝 gpu 版本的 tensorflow 和 keras

安裝完成后,我們使用如下命令,即可檢驗是否成功:

python -c "import keras"

如果看到如下輸出,就說明安裝成功

當然了,我這里安裝的 cuDNN 由於版本過高,暫時不能被pip安裝的tensorflow所支持,如果改為 cuDNN v5 就能夠正常支持了。

希望自己的這篇文章能對像我一樣的新手有所幫助。

參考資料:

[1]: ubuntu16.04下安裝TensorFlow(GPU加速)----詳細圖文教程 

[2]: Ubuntu16.04+cuda8.0+caffe安裝教程 


免責聲明!

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



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