Tensorflow with Gpu 安裝教學

起因

因為教授要求要做物體辨識, 加上實驗室的Server目前暫時不能使用,所以只好先在自己的電腦上面跑

環境

  • Ubuntu 17.04
  • Tensorflow 1.3
  • Cuda 9.0
  • Cudnn 7.0

過程

  1. 安裝 nvidia 的驅動程式

  2. 安裝 cuda - 從官方網站上的 deb 包安裝即可

  3. cudnn 要手動安裝,所以先下載 tar 檔案,

    進到放著tar的資料夾

    用 下面指令解壓縮

    tar -xzvf cudnn-9.0-linux-x64-v7.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*

    ​

  4. 到 /usr/local/cuda/lib64 底下

    使用下面指令, 創造一個軟連結 （因為 Tensorflow 使用 cudnn 6.0, 所以要用這方法把 讓它可以使用 libcudnn.so.6 這個檔案）

    ln -s libcudnn.so.7.* libcudnn.so.6

  5. 用 pip 安裝 tensorflow with Gpu

  6. 在 ~/.bashrc 最底下加上

    export CUDA_HOME=/usr/local/cuda/
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"

選用

如要使用 pycharm 要記得幫它加上環境變數

Written with StackEdit.

留言

張貼留言

這個網誌中的熱門文章

[隨筆] 多執行緒 vs 多進程

圖片
最近因為遇到了這方面的問題，便去求助了一下google大神跟認識的大神 便順手把這篇記錄下來，當作備忘使用 首先，先說一下他們的英文名稱 multi-thread  => 多執行緒 multi-process => 多進程 大致上而言，在排程部分，multi-process是比multi-thread更消耗資源的東東(但也因此速度更快)，因為multi-thread是全部放在一個process裡面，但multi-process則不是，所以在系統排程時會導致multi-process比multi-thread消耗更多資源。 在記憶體部分，multi-process的記憶體是分開的，所以要溝通的話必須調用系統函數而multi-thread因為他的記憶體都是放一起的，所以不用，但也不是說multi-thread沒有缺點，因為如果他同時對同一個區塊做io的話會造成混亂(如下圖所示)，所以這時候便要加上mutex lock來鎖住他，不讓multi-thread能夠同時io，也就是說，當一個記憶體區塊被使用時，其他thread不能去碰它，看是要等他讓cpu空轉還是做其他事情(這也是造成效率高或低的主因，空等太多跟別人比起來會很慢)，此外multi-process是每個process都有一份程式碼，但multi-thread則是所有thread都共用一份，而這在大的程式碼中最明顯，如果記憶體吃不消那就會被放到swap裡面，最後導致所有process都在等swap的io。 此外，如果multi-process其中一個當掉其他的還是能夠正常運行(雖然有些小問題，像是像是orphan process(parent process 當掉)或 zombie process (child process當掉，parent process就成了zombie）)，但是如果multi-process當掉則是全部都被關掉(因為OS在處理時是以process下去看)，舉例而言，就像chrome是多個process所以一個分頁當掉沒差，其他分頁還能正常運作 總而言之 速度方面 multi-process > multi-thread 資源消耗方面 multi-process > multi-thread 記憶體消耗方面 multi-proc
閱讀完整內容

[Onenote]把pdf列印在同一頁

前因 因為想讓Onenote的Pdf可以放在同一頁而不被切割成好幾頁，所以上網稍微找了一下方法 解法 到 檔案 -> 選項 -> 進階 (英文的話會是File -> Options -> Advanced) 找到 在多頁上插入長的列印成品 (Insert long printouts on multiple pages) 並把他取消打勾，這樣就可以了 ^^ 參考 How to print multiple page PDFs onto a single Onenote Page? Written with StackEdit .
閱讀完整內容

Ubuntu 調整硬碟大小教學

圖片
前言 最近在安裝 opencv 時，因為VM硬碟大小不夠，所以只好擴充硬碟大小，但是 root partition 好像沒辦法在開機時調整大小(不確定)，所以這次用 Gparted 過程 首先，先用 live cd 做開機 參考這篇 然後打開 Gparted 會看到被紅色圈起來的部分，然後對她右鍵，按下 swapoff 然後把他移動到右邊(先擴充到右邊，移過去，再縮小)，並且在對他右鍵做swapon 就完成了(灑花花 Written with StackEdit .
閱讀完整內容