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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章