1. 程式人生 > >tensorflow gpu 的安裝

tensorflow gpu 的安裝

1、安裝nvidia-384
安裝完系統之後升級一下系統
sudo apt-get update
sudo apt-get upgrade
ubuntu的官方apt源就有最新最穩定的nv顯示卡驅動。使用ubuntu自己帶的驅動安裝工具直接安裝就可以。
ubuntu-drivers devices

sudo apt-get install nvidia-384

備註也可以使用下載.run等方式安裝,因電腦而異,我的電腦安裝的時候會出現各自各樣的問題所以採取了簡單的方法
提供倆篇文章希望對安裝的過程有幫助
[第一篇連結]該篇文章解決了迴圈登入、重啟黑屏、驅動不工作(

https://www.cnblogs.com/matthewli/p/6715553.html)
第二篇
檢驗 驅動是否正確安裝
cat /proc/driver/nvidia/version
檢驗驅動是否有效工作
nvidia-smi #若列出GPU的資訊列表,表示驅動安裝成功
具體見圖
nvidia-settings #若彈出設定對話方塊,亦表示驅動安裝成功

具體見圖

2、 安裝cuda
a、下載
cuda的下載網址:https://developer.nvidia.com/cuda-downloads?target_os=Linux
sudo dpkg -i cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

b、寫入環境變數
gedit ~/.bashrc
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=:/usr/local/cuda/lib64
c、source
source ~/.bashrc

3、安裝cudnn
a、下載網址https://developer.nvidia.com/cudnn
b、install
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

*4、Installing Bazel*
參考:https://docs.bazel.build/versions/master/install-ubuntu.html
sudo apt-get install openjdk-8-jdk
sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update && sudo apt-get install oracle-java8-installer

echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list

curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -

a、下載bazel
連結:https://github.com/bazelbuild/bazel/releases
特別慢,後續我會給一個我下載好的一一方便後來的使用
b、install
chmod +x bazel--installer-linux-x86_64.sh
./bazel--installer-linux-x86_64.sh –user

c、export
export PATH=”PATH:HOME/bin”

5、安裝tensorflow
a、
將tensorflow放到/home下
執行./configue
配置configure,不懂的就按回車選擇預設方式
b、執行如下命令,在/home/tensorflow
記得加sudo或者在root使用者下執行
命令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
這時在/tmp/tensorflow_pkg/會有一個*.mkl生成
例如:tensorflow-1.5.0rc0-cp27-cp27mu-linux_x86_64.whl

命令4、pip 安裝該包
sudo pip install tensorflow-1.5.0rc0-cp27-cp27mu-linux_x86_64.whl