英偉達(dá)顯卡開(kāi)發(fā)工具CUDA Toolkit

CUDA Toolkit是包含了針對(duì)英偉達(dá)CUDA的軟件開(kāi)發(fā)工具集和編譯器、開(kāi)發(fā)環(huán)境以及各類運(yùn)行庫(kù)，是n卡顯卡CUDA的SDK，如果想要將自己的非圖形計(jì)算程序搬上顯卡去跑，那么這套工具集是必須的。

注意

CUDA依賴Nvidia的驅(qū)動(dòng)，最好先去官網(wǎng)check一下自己的GPU支持的驅(qū)動(dòng)版本和CUDA版本。

安裝CUDA Toolkit

使用--toolkit參數(shù)僅安裝toolkit 或者運(yùn)行安裝程序按指示選擇要安裝的文件。安裝成功后按要求將cuda添加到PATH路徑和LD_LIBRARY_PATH依賴加載路徑中即可。

查看本機(jī)驅(qū)動(dòng)版本

打開(kāi)nvidia的控制面板 -> 系統(tǒng)信息

驅(qū)動(dòng)是cuda9.0版本，安裝cuda10.2時(shí)需要覆蓋此驅(qū)動(dòng)

安裝過(guò)程

選擇自定義安裝，可以選擇安裝驅(qū)動(dòng)，覆蓋本機(jī)的驅(qū)動(dòng)

不要選Visual Studio Integration，即使選了也不能成功安裝

如果本機(jī)的驅(qū)動(dòng)版本(當(dāng)前版本)小于cuda對(duì)應(yīng)的版本（新版本），則選擇，否則不選。如果當(dāng)前版本小于新版本，并且不覆蓋安裝，之后電腦會(huì)頻繁藍(lán)屏或死機(jī)

記住安裝位置，tensorflow要求配置環(huán)境

安裝完成后配置環(huán)境，Path需要手動(dòng)添加如下路徑，對(duì)應(yīng)上一步的安裝路徑
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\bin\win64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\common\lib\x64

相關(guān)新聞

CUDA Toolkit是支持多平臺(tái)的，你在Windows、Linux以及macOS上都可以使用它。NVIDIA與蘋(píng)果之間的關(guān)系裂痕越拉越大了，在去年的macOS Mojave更新中，NVIDIA就遭遇了蘋(píng)果“故意”拖延其新驅(qū)動(dòng)的簽名，導(dǎo)致用戶無(wú)法在新版本macOS上面正常安裝NVIDIA官方顯卡驅(qū)動(dòng)的事件，所以這次僅針對(duì)macOS停止發(fā)布這套工具集可能說(shuō)明NVIDIA與蘋(píng)果之間徹底撕破臉皮。

沒(méi)有了新的CUDA Toolkit，意味著macOS開(kāi)發(fā)者將不能夠在macOS上面使用未來(lái)的CUDA新特性，這兩年的機(jī)器學(xué)習(xí)熱潮讓不少開(kāi)發(fā)者都使用N卡搭建起了自己的機(jī)器學(xué)習(xí)硬件平臺(tái)，可能未來(lái)這些開(kāi)發(fā)者為了CUDA的新特性而不得不放棄界面友好的macOS而轉(zhuǎn)投Windows或Linux了。