1. 程式人生 > >菜鳥安裝Tensorflow

菜鳥安裝Tensorflow

經過樓主的各種掙扎,終於成功地安裝好了tensorflow,參考網上的各種教程遇到很多坑,導致進度緩慢,下面進行最後總結。

首先,在安裝tensorflow之前需要先安裝cuda和cudnn,這個樓主之前並不知道,結果饒了一大圈子。

1.cuda
VIDIA CUDA 解析工具的介面,libcupti-dev 庫。該庫提供了更高階的分析工具支援。要安裝這個庫,對 CUDA Toolkit 8.0 以上的版本執行如下命令:
    sudo apt-get install cuda-command-line-tools
並且將其路徑加在你的環境變數 LD_LIBRARY_PATH 中:
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64
對於 CUDA Toolkit 7.5及以下版本,執行:
    sudo apt-get install libcupti-dev

Ctrl+alt+F1進入字元介面,關閉圖形介面(劃重點!!!不然安裝之後重啟會無法登入圖形介面,只能返回字元介面刪除新安裝的cuda然後再重新來一次!)
    sudo service lightdm stop //必須有,不然會安裝失敗
安裝nvidia driver
    sudo chmod 755 NVIDIA-Linux-x86_64-390.25.run  //獲取許可權  

    sudo ./NVIDIA-Linux-x86_64-390.25.run  //安裝驅動

    sudo service lightdm start

    sudo sh cuda_8.0.61_375.26_linux.run // 安裝過程中的提示輸入:accept,N,Y,Y,Y


接著安裝所缺少的庫:
    sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
    sudo apt-get install vim
設定環境變數
    export PATH=/usr/local/cuda-8.0/bin:$PATH  
    export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
然後修改檔案中環境變數設定
    sudo vi /etc/profile
輸入上面export的兩句,儲存,退出。
    sudo ldconfig //環境變數立即生效
最後,驗證安裝是否完成
    nvidia-smi

2.cudnn 下載
https://developer.nvidia.com/rdp/cudnn-download  (網上不少教程給的連結進入之後的找不到,這個連結登入之後直接跳轉下載頁面)
下載完畢後解壓,然後將相關檔案拷貝到cuda安裝目錄下即可:

tar -zxvf 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/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*


3.tensorflow
一定要注意設定超時時間:(樓主直接在pycharm新建的工程裡面terminal安裝的,tensorflow包含tensorflow和dist-info兩個資料夾)
sudo pip3 --default-timeout=100 install -U tensorflow-gpu==1.4.0


登入PYCHARM:
~/下載/pycharm-2017.3.3/bin$
sh ./pycharm.sh

import tensorflow as tf

如果不報措,那麼就安裝成功啦,開森!!!


寫在文末:

1)如果安裝之後顯示: can not open XXXX file

一般是cuda cudnn tensorflow 相互之間不相容。比如樓主最後確定的版本是 cuda 8.0   cudnn 6.0    tensorflow 1.4。其中遇到cuda 9.0不相容的問題, cudnn5.0不能使用,tensorflow 1.6 需要cuda 9.0 ,一把心酸淚呀!

2)如果安裝tensorflow:

    sudo pip3 install tensorflow-gpu==1.4.0 (結果中途失敗了,其實是安裝超時)

改成:

    sudo pip3 --default-timeout=100 install -U tensorflow-gpu==1.4.0