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

在控制面板中找到NVIDA控制面板，在NVIDA控制面板中，點(diǎn)擊幫助->系統(tǒng)信息->組件

image.png

可以看到本機(jī)支持的是CUDA 11.0 版本，表示是不支持更高版本的。如果你升級(jí)了驅(qū)動(dòng)，可能會(huì)支持更高版本，也可能不會(huì)提升。

所以就必須安裝 11.0 及以下的版本。

下載Cuda

官網(wǎng)：https://developer.nvidia.com/cuda-10.2-download-archive

在https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 這里可以查詢到我們應(yīng)該下載哪個(gè)版本

image.png

安裝過程

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

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

   
   
   image.png
   
   

   之后按步驟安裝。

CUDA安裝完成后，打開powershell，執(zhí)行nvcc -V ，成功的話會(huì)返回cuda版本號(hào)。

下載CuDNN

官網(wǎng) https://developer.nvidia.com/rdp/cudnn-download

image.png

注意：必須選擇和你安裝的CUDA匹配的版本。

解壓cuDNN壓縮包，可以看到bin、include、lib目錄。

找到cuda版本目錄，打開，找到bin、include、lib目錄，將cuDNN壓縮包內(nèi)對(duì)應(yīng)的文件復(fù)制到bin、include、lib目錄。

添加環(huán)境變量

需要添加下面兩個(gè)路徑，這就是說為什么要記住你的安裝路徑了，我使用的是默認(rèn)的安裝路徑。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64

檢查安裝結(jié)果

打開 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite

在此路徑下打開powershell

執(zhí)行：

image.png