CUDA Toolkit是包含了針對英偉達(dá)CUDA的軟件開發(fā)工具集和編譯器、開發(fā)環(huán)境以及各類運(yùn)行庫,是n卡顯卡CUDA的SDK,如果想要將自己的非圖形計(jì)算程序搬上顯卡去跑,那么這套工具集是必須的。
注意
CUDA依賴Nvidia的驅(qū)動,最好先去官網(wǎng)check一下自己的GPU支持的驅(qū)動版本和CUDA版本。
安裝CUDA Toolkit
使用--toolkit參數(shù)僅安裝toolkit 或者運(yùn)行安裝程序按指示選擇要安裝的文件。安裝成功后按要求將cuda添加到PATH路徑和LD_LIBRARY_PATH依賴加載路徑中即可。
查看本機(jī)驅(qū)動版本
打開nvidia的控制面板 -> 系統(tǒng)信息
驅(qū)動是cuda9.0版本,安裝cuda10.2時需要覆蓋此驅(qū)動
安裝過程
選擇自定義安裝,可以選擇安裝驅(qū)動,覆蓋本機(jī)的驅(qū)動
不要選Visual Studio Integration,即使選了也不能成功安裝
如果本機(jī)的驅(qū)動版本(當(dāng)前版本)小于cuda對應(yīng)的版本(新版本),則選擇,否則不選。如果當(dāng)前版本小于新版本,并且不覆蓋安裝,之后電腦會頻繁藍(lán)屏或死機(jī)
記住安裝位置,tensorflow要求配置環(huán)境
安裝完成后配置環(huán)境,Path需要手動添加如下路徑,對應(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是支持多平臺的,你在Windows、Linux以及macOS上都可以使用它。NVIDIA與蘋果之間的關(guān)系裂痕越拉越大了,在去年的macOS Mojave更新中,NVIDIA就遭遇了蘋果“故意”拖延其新驅(qū)動的簽名,導(dǎo)致用戶無法在新版本macOS上面正常安裝NVIDIA官方顯卡驅(qū)動的事件,所以這次僅針對macOS停止發(fā)布這套工具集可能說明NVIDIA與蘋果之間徹底撕破臉皮。
沒有了新的CUDA Toolkit,意味著macOS開發(fā)者將不能夠在macOS上面使用未來的CUDA新特性,這兩年的機(jī)器學(xué)習(xí)熱潮讓不少開發(fā)者都使用N卡搭建起了自己的機(jī)器學(xué)習(xí)硬件平臺,可能未來這些開發(fā)者為了CUDA的新特性而不得不放棄界面友好的macOS而轉(zhuǎn)投Windows或Linux了。