環境配置6-Win10下安裝CUDA和cuDNN,配置caffe
到官網下載相應的安裝包,cuDNN為lib庫,需要先註冊NVIDIA賬號(或者連結複製到迅雷下載)。
https://developer.nvidia.com/rdp/cudnn-archive
https://developer.nvidia.com/cuda-80-ga2-download-archive
1-cuda下載好後,直接按照預設選項安裝。出現下面的提示直接忽略。
2-cudnn下載後解壓(比如解壓後為C:\cuda\),會有bin、include、lib三個資料夾,將這三個資料夾下的內容複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下對應的資料夾中。
(以下3、4、5步驟是否還需要做待驗證,可以先不做,安裝caffe後再觀察)
3-新增cudnn環境變數。
將C:\cuda\bin新增到系統環境變數。
右擊我的電腦——屬性——高階——環境變數——系統變數——PATH
4-開啟visual studio專案
選擇release、x64
除錯——屬性
VC++Directories——Include Directories中新增C:\cuda\include
VC++Directories——Library Directories中新增C:\cuda\lib\x64
Linker——Input——Additional Dependencies中新增cudnn.lib
接下來可以安裝caffe
1-下載Windows下的caffe
https://github.com/BVLC/caffe/tree/windows
2-解壓出來,用notepad++開啟
caffe——scripts——build_win.cmd
3-if DEFINED APPVEYOR不用管,直接看else後的。
MSVC_VERSION=14 (vs2015)
WITH_NINJA=0
CPU_ONLY=0 (如果用GPU)
PYTHON_VERSION=2 (根據安裝的python版本)
其餘的不用改變,儲存後退出。
4-雙擊執行build_win.cmd開始編譯,等待十幾分鍾。
5-編譯成功。將生成的build——lib——release資料夾下的10個檔案,拷貝到程式裡win_lib下替換。