前言
拿到了新本本,就迫不及待想要開始配置好torch的深度學習環境了. 本本是MECHREVO深海系列的Z2-R, CPU i7-9750, 獨立顯卡英偉達RTX-2060 (性價比還算不錯, 再加上本身的輕薄游戲本特性, 對我這種天天碼代碼偶爾打dota2的太合適了). 這篇博客主要記錄CUDA 10.0 + cuDNN 7.5.0的安裝配置流程.
資源下載
- CUDA 10.0 : 官網下載地址 (注意, 如果不掛梯子, 安裝包文件會一直下載失敗, 注意,梯子要用全局模式;網絡安裝版，可以下載，但是無法安裝。)
- cuDNN 7.5.0 : 官網下載地址 (要與CUDA適配版本)
- VS 2015 社區版: 下載地址 (注意, 這個鏈接點擊便直接開始下載了)
- pytorch 1.0 GPU版: 官網下載地址
CUDA安裝
在安裝CUDA的時候提示安裝失敗，在檢索反饋信息之后，我發現 Visual Studio Integration 這個軟件安裝失敗，這個錯誤導致安裝程序無法正常的繼續下去。
解決方案步驟如下:
- 重新安裝CUDA，注意選擇自定義安裝
- 在選擇安裝類別的時候，取消選擇Visual Studio Integration; 並且只安裝CUDA組件, 取消其他的NVIDIA驅動安裝 (可能比你自己電腦的驅動要舊)
- 安裝完成
- 將CUDA安裝包解壓，將CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions 下的所有文件，復制到VS的對應目錄下，參考路徑如下
C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\Common7\IDE\VC\VCTargets\BuildCustomizations
到此為止CUDA就安裝完成了，
( CUDA 9.2: 注意在最后的時候需要運行一下補丁安裝包，這個補丁修復了CUDA9.2某個算法庫的缺陷。)