ubuntu從零開始安裝mxnet--安裝cuda8.0
阿新 • • 發佈:2018-12-17
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安裝檔案
- 安裝輔助的庫
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
- 執行檔案
./cuda_8.0.61_375.26_linux.run
- 執行流程
- 執行一開始會讓我們閱讀一段非常長的協議,我們可以通過ctrl + c跳過這段。並在之後第一個問題輸入accept同意以上協議。
- 不要安裝nvidia驅動!!!!(關鍵),在問題中輸入n
- 接下來我們只需要輸入yes和保持預設路徑即可
- 安裝成功
檢查和補充安裝
- 執行
nvcc -V
會出現兩種情況。
- 第一種是我們之前的安裝沒有完全安裝完成,會有如下提示
- 我們只需要根據提示執行
sudo apt install nvidia-cuda-toolkit
即可,不過可能時間有點久,請耐心等待。
- 我們只需要根據提示執行
- 經過一番耐心等待之後,安裝完成,再次輸入
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