windows下caffe-ssd（GPU版）搭建测试环境

一、准备CUDA、CuDNN

CUDA下载：https://developer.nvidia.com/cuda-downloads (注意CUDA的版本与VS的版本匹配）

CuDNN下载：https://developer.nvidia.com/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一下，继续编译

三、可以开始训练了

参考：https://blog.csdn.net/Rayue_/article/details/81080359