到官網下載相應的安裝包,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下替換。