1. 程式人生 > >ubuntu從零開始安裝mxnet--安裝cuda8.0

ubuntu從零開始安裝mxnet--安裝cuda8.0

CUDA是一種由NVIDIA推出的通用平行計算架構,該架構使GPU能夠解決複雜的計算問題。
根據mxnet官網的說法,我們應該安裝的是cuda8.0的版本。

下載cuda

這裡筆者給出了最新版cuda的下載地址,https://developer.nvidia.com/cuda-downloads,8.0版本的下載地址不明原因無法開啟。筆者給出了百度雲的地址,不過只有Ubuntu16.04 cuda8.0.61版本。連結: https://pan.baidu.com/s/1nuCY4yd 密碼: jcrj
如果小夥伴自行下載請選擇.run檔案進行下載,因為其他方式,筆者進行了多次嘗試都很難成功。

驅動準備

安裝nvidia顯示卡驅動,可以看一下筆者的上一篇文章《ubuntu從零開始安裝mxnet–安裝NVIDIA驅動》

執行cuda安裝檔案

  1. 安裝輔助的庫
    sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
  2. 執行檔案
    ./cuda_8.0.61_375.26_linux.run
  3. 執行流程
    • 執行一開始會讓我們閱讀一段非常長的協議,我們可以通過ctrl + c跳過這段。並在之後第一個問題輸入accept同意以上協議。
    • 不要安裝nvidia驅動!!!!(關鍵),在問題中輸入n
      這裡寫圖片描述
    • 接下來我們只需要輸入yes和保持預設路徑即可
    • 安裝成功
      這裡寫圖片描述

檢查和補充安裝

  • 執行nvcc -V會出現兩種情況。
    1. 第一種是我們之前的安裝沒有完全安裝完成,會有如下提示
      這裡寫圖片描述
      • 我們只需要根據提示執行sudo apt install nvidia-cuda-toolkit即可,不過可能時間有點久,請耐心等待。
    2. 經過一番耐心等待之後,安裝完成,再次輸入nvcc -V,得到如下結果
      這裡寫圖片描述

執行一個例子

在安裝完成後,我們可以嘗試執行官方給的一個sample,如果我們之前安裝的時候按照預設路徑安裝的話,不出意外會在/root/NVIDIA_CUDA-8.0_Samples資料夾下。
1. 執行cd /root/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery進入資料夾
2. 執行make編譯檔案
3. 執行./deviceQuery,如果出現了Result = PASS字樣,恭喜,之前的步驟全部成功,可以進行下一步。
這裡寫圖片描述

新增環境變數

編輯vim /etc/profile,在最後面新增

export PATH=$PATH:/usr/local/cuda-8.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-8.0/lib64

最後執行source /etc/profile