NVIDIA CUDA 9.0是英偉達(dá)開(kāi)發(fā)的一個(gè)并行計(jì)算平臺(tái),這款工具需要英偉達(dá)的獨(dú)立顯卡支持。使用CUDA Toolkit,您可以在GPU加速的嵌入式系統(tǒng),桌面工作站,企業(yè)數(shù)據(jù)中心,基于云的平臺(tái)和HPC超級(jí)計(jì)算機(jī)上開(kāi)發(fā),優(yōu)化和部署應(yīng)用程序。該工具包包括GPU加速庫(kù),調(diào)試和優(yōu)化工具,C / C ++編譯器和用于部署應(yīng)用程序的運(yùn)行時(shí)庫(kù)。
軟件功能:
CUDA是一個(gè)由NV公司發(fā)明的并行計(jì)算平臺(tái)和編程模型。
它利用圖形處理單元(GPU)的強(qiáng)大功能,使計(jì)算性能大幅提高。
提供向C這樣的標(biāo)準(zhǔn)語(yǔ)言的擴(kuò)展,從而實(shí)現(xiàn)并行算法的簡(jiǎn)單實(shí)現(xiàn)。
通過(guò)使用CUDA C/C++可以是編程人員專注于算法的并行化而不是程序本身的實(shí)現(xiàn)
支持應(yīng)用程序同時(shí)使用CPU和GPU進(jìn)行異構(gòu)計(jì)算。
程序的串行部分運(yùn)行在CPU上,并行部分則運(yùn)行在GPU上。
安裝方法:
因?yàn)樽钚掳娴腡ensorflow_gpu兼容CUDA9.0的版本容易從網(wǎng)上獲取,即通過(guò)pip install tensorflow_gpu安裝的版本是可以兼容CUDA9.0的。
選擇下圖中紅色箭頭標(biāo)注處exe可執(zhí)行文件,雙擊開(kāi)始安裝。
不要修改默認(rèn)安裝位置,直接點(diǎn)擊OK,如下圖中紅色箭頭標(biāo)注處所示。
如下圖所示,正在提取安裝文件中的內(nèi)容,等待即可。
上圖運(yùn)行完成后,回自動(dòng)彈出NVIDIA安裝程序,如下圖所示。
在檢測(cè)系統(tǒng)兼容性這一步,等待即可。
本文作者的顯卡是Nvidia RTX2070,在下圖中安裝程序顯示沒(méi)有發(fā)現(xiàn)可以兼容的顯卡設(shè)備。
因?yàn)镃UDA9.0比RTX2070更早發(fā)布,所以沒(méi)有找到是合理的,但是安裝后仍然可以正常使用。
本文讀者可能直接跳到“許可協(xié)議”這一步,則說(shuō)明安裝程序找到了可兼容的顯卡設(shè)備。
點(diǎn)擊下圖紅色箭頭標(biāo)注處,進(jìn)入下一步。
許可協(xié)議不用仔細(xì)去閱讀,點(diǎn)擊下圖紅色箭頭標(biāo)注處,進(jìn)入下一步。
安裝模式選擇精簡(jiǎn),點(diǎn)擊下圖紅色箭頭標(biāo)注處,進(jìn)入下一步。
勾選下圖中的上方紅色方框標(biāo)注處,然后NEXT按鈕會(huì)亮起。
點(diǎn)擊下圖中的下方紅色箭頭標(biāo)注處中的NEXT按鈕,進(jìn)入下一步。
安裝界面顯示Preparing for installation,即正在準(zhǔn)備安裝。
安裝過(guò)程中部分截圖如下圖所示。
安裝完成后,安裝程序界面如下圖所示。
點(diǎn)擊下圖中的下方紅色箭頭標(biāo)注處中的下一步。
結(jié)束界面中兩個(gè)可選項(xiàng)不用勾選。
點(diǎn)擊下圖中的下方紅色箭頭標(biāo)注處中的關(guān)閉,安裝CUDA9.0結(jié)束。