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