煉丹爐(2)--系統安裝:Ubuntu 16.10 + cuda 8.0 + cudnn 5.1 + tensorflow r1.0 GPU

裝機硬體配置: 煉丹爐(1)--8000~1.2 萬的台式機選擇 + 100 元裝機費(裝機小白)

選擇 ubuntu 16.10 而非 16.04 的原因見:煉丹爐(深度學習)上裝 ubuntu 系統的幾個坑

cuda 和 cudnn 是 Nvidia 出的 GPU 加速工具。

GPU 版本的 tensorflow,依賴於這兩個工具。

安裝 cuda 和 cudnn 時,需要先在 Nvidia 官網下載安裝包。

免費註冊賬號以後,才可以下載。

關於 gcc 的版本。

cuda 官網以及各種攻略里,都提到 gcc 版本太新不支持。

其實,那是針對基於 cuda 開發者的。

我們只裝個 tensorflow,對 gcc 版本沒要求。

gcc 版本不對,cuda 的 examples 肯定是編譯不通過的。

然而,不在乎。

tensorflow 跑的很正常。

安裝 ubuntu 16.10

首先,要製作 ubuntu 啟動 U 盤。不細說。

安裝期間,我的無線滑鼠不能用,安裝後正常。

應該是驅動的問題。

用鍵盤操作可以完成安裝。

關於 swap 分區

建議,設為內存大小的 1.5 倍。

help.ubuntu.com 上關於 swap 分區的詳細討論:SwapFaq - Community Help Wiki

askubuntu.com 針對 16G 內存如何劃分 swap 分區的問題 I have 16GB RAM. Do I need 32GB swap?

換一個速度快的 source

安裝 CUDA 8.0

用的 deb(network) 的安裝方式

官網的下載地址:developer.nvidia.com/cu

下面的 wget 命令就是下載 deb(network) 安裝包用的。

$ wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.debn$ sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.debn$ sudo apt-get updaten$ sudo apt-get install cudan

要下載 1.4G 的安裝文件,所以,有點慢。

設置環境變數:

在 ~/.bashrc 的最後添加

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}nexport LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}nexport CUDA_HOME=/usr/local/cudan

其中,

前 2 個(PATH, LD_LIBRARY_PATH) 是 CUDA 官網安裝文檔中建議的變數。

第 3 個(CUDA_HOME)是 tensorflow-GPU 版本要求的變數。

重啟電腦,否則顯卡驅動不生效。

檢查顯卡驅動安裝結果

$ nvidia-smin

安裝 cudnn 5.1

官網的下載地址:developer.nvidia.com/rd

選擇:cuDNN v5.1 Library for Linux

$ tar xvzf cudnn-8.0-linux-x64-v5.1.tgzn$ sudo cp cuda/include/cudnn.h /usr/local/cuda/includen$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64n$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*n

安裝 tensorflow r1.0 GPU 版本

見官網:tensorflow.org/install/

命令匯總如下

$ sudo apt-get install libcupti-devn$ sudo apt-get install python-pip python-dev python-virtualenvn$ virtualenv --system-site-packages ~/tensorflow-gpun# 開 VPN 下載 whl 格式的安裝包,速度快。n$ wget https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.0-cp27-none-linux_x86_64.whln$ source ~/tensorflow-gpu/bin/activaten# 從豆瓣的鏡像安裝其它依賴包n(tensorflow-gpu)$ pip install tensorflow_gpu-1.0.0-cp27-none-linux_x86_64.whl -i http://pypi.douban.com/simple/ --trusted-host=pypi.douban.comn

驗證 tensorflow 安裝

進入 python 的互動式命令行

$ pythonn

執行下列代碼

>>> import tensorflow as tfn>>> hello = tf.constant(Hello, TensorFlow!)n>>> sess = tf.Session()n>>> print(sess.run(hello))n

預期輸出:

Hello, TensorFlow!n


推薦閱讀:

煉丹爐(3)--常用軟體安裝
Ubuntu 16.04 體驗如何?
Ubuntu 18.04 LTS 定名為「仿生河狸」,將於明年 4 月 26 日發布
Ubuntu16.04卡在開機畫面問題的解決記錄
Win+Ubuntu雙系統下Nvidia獨顯驅動安裝

TAG:深度学习DeepLearning | GPU通用计算 | Ubuntu |