學習深度學習首先須要搭建開發環境,如今深度學習框架有不少,可是我仍是選擇google開源的tensorflow,安裝相對容易一點,適合初學者。因爲我是作研究用的,因此我下血本配置了一臺GTX1060-6g的顯卡。因爲個人電腦有顯卡,因此我選擇了裝支持gup版本的tensorflow。若是不須要裝支持gpu版本的能夠參考 http://blog.csdn.net/keith_bb/article/details/74066691這篇文章來安裝。因爲我買的帶有GPU的本子,我想不能浪費個人本子,因此我爬了無數坑裝好了支持gup版本的tensorflow。下面我就分享一下個人安裝過程。本次是用源碼來進行安裝的。html
環境框架:Ubuntu16.04+cuda8.0+cudnn v6+tensorflow1.4java
step1:python
下載所需的軟件包。linux
一、下載系統鏡像文件Ubuntu16.04 地址https://www.ubuntu.com/download/alternative-downloadsc++
如圖所示下載鏡像文件。git
二、下載所需的cuda8.0(gtx1060也支持cuda8.0)github
地址https://developer.nvidia.com/cuda-downloadsweb
如今官網直接更新到9.0ubuntu
cuda 8.0地址https://developer.nvidia.com/cuda-80-ga2-download-archiveapi
三、下載cudnn v6
下載地址:https://developer.nvidia.com/cudnn
下載時可能須要登陸,註冊帳號後填寫調差問卷就能夠下載。
下載好所須要的安裝包就能夠進行按住安裝了。
step2:安裝
一、首先安裝顯卡GPU驅動
打開終端:sudo apt-get update
選擇系統設置→軟件更新→附加驅動→選擇nvidia最新驅動→應用更改.
二、安裝Tensorflow依賴的編譯工具bazel
bazel安裝方法網址:https://bazel.build/versions/master/docs/install-ubuntu.html
(1) 安裝bazel前,需先安裝JDK8
sudo apt-get installsoftware-properties-common
sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update
sudo apt-get installoracle-java8-installer
驗證java版本:java -version
(2) 安裝bazel
echo "deb [arch=amd64]http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee/etc/apt/sources.list.d/bazel.list
sudo apt install curl
curlhttps://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
sudo apt-get update
sudo apt-get upgrade bazel
三、因爲cuda8.0不支持gcc 5.0以上的編譯器,所以須要降級,把編譯器版本降到4.9:
先查看本身的gcc版本
gcc –v
下降gcc版本:
sudoapt-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++
查看gcc版本
四、安裝cuda 8.0
切換目錄到cuda源文件下:
sudo dpkg -icuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
五、安裝cuDNN 6.0
cp cudnn-8.0-linux-x64-v6.0.solitairetheme8 cudnn-8.0-linux-x64-v6.0.tgz
tar -xvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
六、配置環境變量
sudo gedit ~/.bashrc
export LD_LIBRARY_PATH=」$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64」
export CUDA_HOME=/usr/local/cuda
export PATH="$CUDA_HOME/bin:$PATH"
繼續在terminal中輸入:
source ~/.bash_profile #使更改的環境變量生效
七、安裝Tensflow
(1) 安裝Tensorflow依賴的其它工具包
sudo apt-get install python-numpy swigpython-dev python-wheel
(2) 下載最新的Tensorflow源碼
sudo apt-get install git
git clone https://github.com/tensorflow/tensorflow
(3) 運行configure腳本配置環境信息(一路按回車建默認就行)或者參照下圖配置。
八、 經過bazel來編譯pip的安裝包,而後經過pip安裝
在terminal中輸入如下命令:(仍是在tensorflow目錄下)
Traceback (most recent call last): File "setup.py", line 25, in <module> from setuptools import find_packages, setup, Command ImportError: No module named setuptools setuptools-18.1https://www.cnblogs.com/huanbia/p/4725421.html 1.bazel build -c opt //tensorflow/tools/pip_package:build_pip_package 2.bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 3.bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg(編譯後的安裝包在這) 此處可能會出現以下問題:直安裝參考連接4.sudo pip install /home/***(你本身的用戶名)/Desktop/tensorflow-1.4.0-cp2-none-any.whl
上述第4部可能會出錯,須要安裝pip 包
命令:
sudo apt-get install python-pip
以後執行第四部代碼效果以下:
如圖所示 安裝成功!!!!