CUDA 簡介
CUDA 是由英偉達(dá)公司開發(fā)的并行計算平臺和編程模型的統(tǒng)稱。通過這個平臺我們可以利用 GPU 并行計算的能力來加速一些復(fù)雜的計算。
CUDA 有兩個特性:一是其編程語言由現(xiàn)在流行的語言拓展而來(比如C、C++)甚至也可以用 Python 來開發(fā),這大大減少了開發(fā)人員上手的難度;二是支持混合計算,簡單地說就是在我們一個程序中同時擁有 CPU 執(zhí)行的順序計算部分和 GPU 執(zhí)行的并行計算部分。
那么,下面我們就開始 Ubuntu 下 CUDA 環(huán)境搭建。
在安裝之前
系統(tǒng)要求
要搭建 CUDA 環(huán)境,我們需要自己的計算機(jī)滿足以下這三個條件:
1. 有至少一顆支持 CUDA 的 GPU(我的是GeForece GT 650M)
2. 有滿足版本要求的 gcc 編譯器和鏈接工具
3. 有 NVIDIA 提供的 CUDA 工具包(這個可以到http://developer.nvidia.com/cuda-downloads下載)
準(zhǔn)備工作
下面,我們一步一步來驗證自己的系統(tǒng)是否滿足安裝要求。
一、驗證計算機(jī)是否擁有至少一顆支持 CUDA 的 GPU
打開終端(Ctrl?+ Alt + t),鍵入以下命令:
lspci | grep -i nvidia
可以看到以下內(nèi)容(結(jié)果會與我不同,跟你的 GPU 有關(guān))
看到這個就說明有一顆支持 CUDA 的 GPU,可以進(jìn)入下一步了。
二、驗證一下自己操作系統(tǒng)的版本
鍵入命令:
uname -m && cat /ect/*release
可以看到系統(tǒng)的版本是 Ubuntu14.04 并且是64位系統(tǒng)(這個很重要,要下載對應(yīng)系統(tǒng)的安裝包)。
三、驗證 gcc 編譯器的版本
鍵入命令:
gcc --version
四、驗證系統(tǒng)內(nèi)核版本
鍵入命令:
uname -r
五、最后附上一張官方提供的對各種 Linux 發(fā)行版的安裝要求
搭建 CUDA 環(huán)境
一、安裝 CUDA 工具包
在前面幾項驗證都順利通過以后就來到最關(guān)鍵的一步。首先下載對應(yīng)自己系統(tǒng)版本的 CUDA 工具包(我的是 cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb),然后進(jìn)入到安裝包所在目錄,我的放在主目錄所以執(zhí)行:
cd ~
在此目錄下執(zhí)行:
sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
此時一般會提示輸入密碼,輸入自己的密碼就行了。
再執(zhí)行:
sudo apt-get update
以及:
sudo apt-get install cuda
此時靜靜地等待安裝完成。不出意外,一段時間后安裝完成了。
二、設(shè)置環(huán)境變量
首先在 PATH 變量中加入 /usr/local/cuda-7.5/bin,在終端中執(zhí)行:
export PATH=/usr/local/cuda-7.5/bin:$PATH
然后在 LD_LIBRARY_PATH 變量中添加 /usr/local/cuda-7.5/lib64,執(zhí)行:
export? LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
三、驗證環(huán)境搭建是否成功
首先執(zhí)行命令:
nvcc -V
看到以下結(jié)果:
再進(jìn)入各種例程所在目錄,執(zhí)行:
cd ~/NVIDIA_CUDA-7.5_Samples
在此目錄下,執(zhí)行:
make
此時,會將官方提供的例程編譯成可執(zhí)行文件,并且放到 ~/NVIDIA_CUDA-7.5_Samples/bin這個目錄下,執(zhí)行:
cd ~/NVIDIA_CUDA-7.5_Samples/bin/x86_64/linux/release
進(jìn)入這個目錄后,運(yùn)行 deviceQuery 這個程序:
./deviceQuery
可以看到通過測試。到這里,64位 Ubuntu?14.04 系統(tǒng)下 CUDA 環(huán)境搭建就完成了。