一、準備CUDA、CuDNN

CUDA安裝：

1、雙擊下載好的exe檔案，選擇路徑，開始安裝

2、系統檢查

3、同意條款並繼續

4、選擇自定義安裝

5、記好下面三個安裝位置，一直下一步，直到安裝完成

CuDNN安裝：

1、解壓下載下來的CuDNN包，裡面有一個cuda資料夾，進入資料夾

2、將其中include的下的cudnn.h檔案複製到 CUDA Toolkit 安裝目錄下的對應位置

將lib/x64下的cudnn.lib複製到CUDA Toolkit 安裝目錄下的對應位置

將bin下的cudnn64_7.dll複製到CUDA Toolkit 安裝目錄下的對應位置

二、CMake後編譯

1、新建資料夾caffe-ssd-gpu，開啟cmake

2、選擇vs版本，configure

3、報錯、修改

將BLAS選項改為OPEN、確保python、python_layer選項選中、USE_CUDNN選中

configure，保證以下結果正確

4、Generate + Open Project

5、將ALL_BUILD設為啟動專案（右鍵->設為啟動專案），在debug模式下編譯（右鍵->重新生成）

編譯成功後換成release模式下編譯

然後將ISTALL設為啟動專案在release模式下編譯

6、你可能會遇到問題  Unsupported gpu architecture 'compute_xx'

開啟caffe-ssd-window-master/cmake資料夾下的Cuda.cmake檔案（寫字板開啟）

把對應的錯誤編號刪掉就行，如：我的就改成  set(Caffe_known_gpu_archs "30 35 50 60 61")

然後重新cmake一下，繼續編譯

三、可以開始訓練了