卸載CUDA(以前安裝過cuda,必須要卸載干凈)
1. 首先是軟件卸載,按照名字排序,以下兩個紅框內的別卸載,其他全部卸載。有的教程說GeForce Experience也別卸載,我卸載了也沒啥影響。
2. 使用殺毒軟件請理注冊表。
3. 刪除三個文件夾【C:\ProgramData】,【C:\Program Files】,【C:\Program Files (x86)】中的“NVIDIA Corporation”和“NVIDIA GPU Computing Toolkit”。
4. 如果文件無法刪除,打開服務管理器(win10搜索“計算機管理”),停止以下服務。
5. 再將3中文件清理干凈,重啟電腦。
安裝Visual Studio 2019
不安裝VS,有些功能無法正常使用。選擇C++的桌面開發
安裝CUDA
1. 先查看電腦支持的版本,打開NIVDIA控制面板,菜單欄->幫助->系統信息->組件
看到是11.0版本,好像是11.0及以下都可以的,但我還是選擇裝11.0。
2. 打開官網(https://developer.nvidia.com/cuda-toolkit-archive),選擇11.0
網頁下載很慢,使用迅雷下載local版本
3. 運行安裝包,直接安裝精簡版本
4. Path環境變量添加
5. 測試,打開cmd
nvcc -V
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\demo_suite
bandwidthTest.exe
deviceQuery.exe
安裝cudnn
1. 官網下載(https://developer.nvidia.com/cudnn),登錄賬號,做問卷
2. 將解壓文件里面的bin、include、lib文件夾復制到CUDA安裝目錄C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
安裝pytorch
pytorch官網(https://pytorch.org/get-started/locally/)可以獲得安裝命令
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
下載可能比較慢,解決方法
換源 或 下載離線安裝包(參考https://www.cnblogs.com/farewell-farewell/p/11096438.html)