CUDA 入門之 Ubuntu 系統(tǒng)下環(huán)境搭建

CUDA 簡(jiǎn)介

CUDA 是由英偉達(dá)公司開發(fā)的并行計(jì)算平臺(tái)和編程模型的統(tǒng)稱。通過這個(gè)平臺(tái)我們可以利用 GPU 并行計(jì)算的能力來加速一些復(fù)雜的計(jì)算。

CUDA 有兩個(gè)特性:一是其編程語言由現(xiàn)在流行的語言拓展而來(比如C、C++)甚至也可以用 Python 來開發(fā),這大大減少了開發(fā)人員上手的難度;二是支持混合計(jì)算,簡(jiǎn)單地說就是在我們一個(gè)程序中同時(shí)擁有 CPU 執(zhí)行的順序計(jì)算部分和 GPU 執(zhí)行的并行計(jì)算部分。

那么,下面我們就開始 Ubuntu 下 CUDA 環(huán)境搭建。

在安裝之前

系統(tǒng)要求

要搭建 CUDA 環(huán)境,我們需要自己的計(jì)算機(jī)滿足以下這三個(gè)條件:

1. 有至少一顆支持 CUDA 的 GPU(我的是GeForece GT 650M)

2. 有滿足版本要求的 gcc 編譯器和鏈接工具

3. 有 NVIDIA 提供的 CUDA 工具包(這個(gè)可以到http://developer.nvidia.com/cuda-downloads下載)

準(zhǔn)備工作

下面,我們一步一步來驗(yàn)證自己的系統(tǒng)是否滿足安裝要求。

一、驗(yàn)證計(jì)算機(jī)是否擁有至少一顆支持 CUDA 的 GPU

打開終端(Ctrl?+ Alt + t),鍵入以下命令:

lspci | grep -i nvidia

可以看到以下內(nèi)容(結(jié)果會(huì)與我不同,跟你的 GPU 有關(guān))

看到這個(gè)就說明有一顆支持 CUDA 的 GPU,可以進(jìn)入下一步了。

二、驗(yàn)證一下自己操作系統(tǒng)的版本

鍵入命令:

uname -m && cat /ect/*release

可以看到系統(tǒng)的版本是 Ubuntu14.04 并且是64位系統(tǒng)(這個(gè)很重要,要下載對(duì)應(yīng)系統(tǒng)的安裝包)。

三、驗(yàn)證 gcc 編譯器的版本

鍵入命令:

gcc --version

四、驗(yàn)證系統(tǒng)內(nèi)核版本

鍵入命令:

uname -r

五、最后附上一張官方提供的對(duì)各種 Linux 發(fā)行版的安裝要求

搭建 CUDA 環(huán)境

一、安裝 CUDA 工具包

在前面幾項(xiàng)驗(yàn)證都順利通過以后就來到最關(guān)鍵的一步。首先下載對(duì)應(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

此時(shí)一般會(huì)提示輸入密碼,輸入自己的密碼就行了。

再執(zhí)行:

sudo apt-get update

以及:

sudo apt-get install cuda

此時(shí)靜靜地等待安裝完成。不出意外,一段時(shí)間后安裝完成了。

二、設(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

三、驗(yàn)證環(huán)境搭建是否成功

首先執(zhí)行命令:

nvcc -V

看到以下結(jié)果:

再進(jìn)入各種例程所在目錄,執(zhí)行:

cd ~/NVIDIA_CUDA-7.5_Samples

在此目錄下,執(zhí)行:

make

此時(shí),會(huì)將官方提供的例程編譯成可執(zhí)行文件,并且放到 ~/NVIDIA_CUDA-7.5_Samples/bin這個(gè)目錄下,執(zhí)行:

cd ~/NVIDIA_CUDA-7.5_Samples/bin/x86_64/linux/release

進(jìn)入這個(gè)目錄后,運(yùn)行 deviceQuery 這個(gè)程序:

./deviceQuery

可以看到通過測(cè)試。到這里,64位 Ubuntu?14.04 系統(tǒng)下 CUDA 環(huán)境搭建就完成了。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容