需要注意的前提條件
我做了這幾個先驗條件,全程安裝沒有踩到bug,很順利~
1. 需要卸載顯卡驅動:link
2. 有以前的CUDA版本的話需要卸載,在控制面板-卸載程序中,
清除帶有CUDA的所有東西、
再清除除了NVIDIA 圖形驅動程序、NVIDIA PhysX 系統軟件、NVIDIA GeForce Experience(只保留這三個)之外的NVIDIA的東西
3.需要刪除以前的VS版本,這裏是VS2019來裝的,卸載我沒有做,可以參考link
4.此外,還需要查詢GPU版本是否能夠被CUDA所支持。
1.VS2019安裝
進入官網:VS2019
點擊Visual Studio2019,我下載的Community版本,一路點擊,安裝後打開installer。
注意:只選擇 使用C++桌面開發 這個選項,基本就能滿足要求,我是安裝的默認路徑
安裝之後重啓,進入CUDA安裝部分:
2.CUDA10.2安裝
進入https://developer.nvidia.com/cuda-downloads
選擇相應的CUDA版本進行安裝。
在安裝中,注意選擇自定義,勾選所有選項,一路點擊即可。(這一部分會自動安裝顯卡驅動)
這裏安裝它自動添加了環境變量,可以到高級系統變量-PATH中查得到。
測試:在cmd中輸入nvcc -V查看是否有版本信息,有則安裝成功
3.CUDNN
https://developer.nvidia.com/rdp/cudnn-download
勾選I agree 下載對應的版本
解壓後是三個文件夾,將其粘貼到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2文件夾中。
測試
至此,安裝完畢
可以通過在python環境中(anaconda虛擬環境也OK)
import torch
temp=torch.Tensor([3.])
temp.cuda()
若不報錯則成功安裝。
借鑑blog:
https://blog.csdn.net/houzupi2534/article/details/100010836
http://www.freesion.com/article/2481243398/
https://www.cnblogs.com/guoyaohua/p/9265268.html