首先要明確gpu是哪個(gè)顯卡并且已安裝的顯卡驅(qū)動(dòng)或者將要安裝哪個(gè)版本，nVidia官網(wǎng)https://www.nvidia.com/Download/index.aspx，

?

1. 安裝CUDA10.1

① 先查看本機(jī)顯卡適合哪個(gè)版本的CUDA（鄰近版本有時(shí)也可以，如果項(xiàng)目需求和電腦要求不一樣，可以嘗試一下可不可以），打開(kāi)“控制面板”---》“硬件與聲音”---》“NVIDIA控制面板”---》“幫助”----》“系統(tǒng)信息”--->“組件”，如圖二紅線所示，我系統(tǒng)適合CUDA10.1。

注：

最主要還是去nVidia官網(wǎng)?https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html?查看cuda和顯卡驅(qū)動(dòng)對(duì)應(yīng)表

② 知道適合哪個(gè)版本的CUDA，就可以去官網(wǎng)下載對(duì)應(yīng)的CUDA了，但是官網(wǎng)首頁(yè)的CUDA一般是最新版，我們可能需要下載舊版本，https://developer.nvidia.com/cuda-toolkit-archiv，進(jìn)入這個(gè)網(wǎng)址下載可以看到舊版本列表，如下圖所示。

?

點(diǎn)擊相應(yīng)版本，進(jìn)入下圖界面，選擇OS，系統(tǒng)64位，版本，然后選擇exe(local)是一個(gè)完整的包，exe(network)得自己安裝組件，下載完CUDA，然后進(jìn)行安裝就行。點(diǎn)擊相應(yīng)版本，進(jìn)入下圖界面，選擇OS，系統(tǒng)64位，版本，然后選擇exe(local)是一個(gè)完整的包，exe(network)得自己安裝組件，下載完CUDA，然后進(jìn)行安裝就行，安裝完成，在win+R輸入cmd打開(kāi)命令行輸入 nvcc -V，如下圖所示安裝成功。

注：如果CUDA是自定義安裝，一般系統(tǒng)不會(huì)自動(dòng)給CUDA配置環(huán)境變量，需要手動(dòng)配置，此時(shí)，nvcc -V命令沒(méi)有正常輸出。通過(guò)在系統(tǒng)變量path中添加CUDA目錄下的bin和libnvvp兩個(gè)目錄即可，如下圖。

?

2. 安裝cudnn

① 下載cudnn，https://developer.nvidia.com/rdp/cudnn-archive?進(jìn)入各個(gè)舊版本的cudnn的網(wǎng)址，選擇相對(duì)應(yīng)CUDA的cudnn版本，一個(gè)CUDA有多個(gè)cudnn版本對(duì)應(yīng)，隨便選一個(gè)即可

② 下載后，將壓縮包解壓得到cuda文件夾，文件夾下有三個(gè)文件夾，復(fù)制這三個(gè)文件夾到目錄C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1下，會(huì)自動(dòng)將cudnn的三個(gè)文件夾的文件合并到CUDA\v10.1的三個(gè)同名文件夾bin、include和lib中，如下圖所示。

3. 安裝Pytorch

https://pytorch.org/get-started/locally/，進(jìn)入pytorch官網(wǎng)進(jìn)行系統(tǒng)，語(yǔ)言，安裝方式的選擇進(jìn)行安裝。要想得到安裝其它舊版本的命令，可以點(diǎn)擊下圖中藍(lán)色線處轉(zhuǎn)到舊版本頁(yè)面。

推薦使用pip安裝，如果有安裝annaconda，選擇conda就行，然后選擇相應(yīng)CUDA版本，得到了下圖紅線所示的命令，在Windows命令行輸入命令即可。

?

pytorch舊版本頁(yè)面

?

注意：雖然圖中pytorch1.5.1可以匹配CUDA10.1，但是我安裝pytorch1.5.1和pytorch1.4.0都沒(méi)有安裝成功，會(huì)有下圖所示的錯(cuò)誤。

?

最后安裝pytorch1.3.0成功。但是cuda9.2版本的pytorch，我前面安裝的cuda是10.1，所以如果官網(wǎng)的安裝命令和版本有很多莫名其妙問(wèn)題的時(shí)候，可以自己多嘗試其它的安裝版本。如直接在pip install時(shí)指定torch==1.3.0來(lái)指定版本，然后torchvision的話可以安裝torch時(shí)自動(dòng)安裝，若沒(méi)有自動(dòng)安裝，則手動(dòng)pip install安裝。

注：

手動(dòng)安裝可以直接轉(zhuǎn)去pytorch的whl網(wǎng)頁(yè)https://download.pytorch.org/whl/cu101/torch_stable.html，比如cuda10.1版本就去cu101，cuda9.2版本就去cu92網(wǎng)頁(yè)，然后下載whl文件后，直接pip install torch-1.1.0-whl（改成你的文件名）就行。

?

?

簡(jiǎn)單測(cè)試是否安裝成功，可通過(guò)在命令行或終端import torch這個(gè)包

?

然后最終測(cè)試是否成功可通過(guò)簡(jiǎn)單運(yùn)行以下代碼測(cè)試

總結(jié)

以上是生活随笔為你收集整理的Windows 10安装CUDA10.1+cudnn7.6.0+Pytorch1.3.0的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。