煉丹爐(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 倍。
http://help.ubuntu.com 上關於 swap 分區的詳細討論:SwapFaq - Community Help Wiki
http://askubuntu.com 針對 16G 內存如何劃分 swap 分區的問題 I have 16GB RAM. Do I need 32GB swap?
換一個速度快的 source
安裝 CUDA 8.0用的 deb(network) 的安裝方式
官網的下載地址:https://developer.nvidia.com/cuda-downloads
下面的 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
官網的下載地址:https://developer.nvidia.com/rdp/cudnn-download
選擇: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 版本
見官網:https://www.tensorflow.org/install/install_linux
命令匯總如下
$ 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 |