VS2015 编译windows caffe gpu

caffe-windows/scripts/build_win.cmd

大概81 line:

(1)

    :: [Fermi  Kepler  Maxwell  Pascal  All]
    if NOT DEFINED CUDA_ARCH_NAME set CUDA_ARCH_NAME=All

(2)

python 我设置成0，不编译python；需要可以打开，需要安装python;

 if NOT DEFINED PYTHON_VERSION set PYTHON_VERSION=3
    :: Change these options for your needs.
    if NOT DEFINED BUILD_PYTHON set BUILD_PYTHON=0
    if NOT DEFINED BUILD_PYTHON_LAYER set BUILD_PYTHON_LAYER=0

(3) cuda

 -DCUDA_ARCH_NAME:STRING=%CUDA_ARCH_NAME% ^
      -DCUDNN_ROOT=F:/project/Code/caffe/bag/caffe-windows/cudnn-8.0-windows10-x64-v5.1/cuda ^

(4)注意

caffe-windows/cmake/Cuda.cmake

要支持的计算能力查看：

set(Caffe_known_gpu_archs "20 21(20) 30 35 50 60 61")

如果要支持上面，需要在caffe-windows/scripts/build_win.cmd 设置：

set CUDA_ARCH_NAME=All

(5)cuda用8.0;  vs2015; cudnn:5.1（ for cuda 8.0）;

双击编译，注意查看cuda和cudnn版本是否正确；

编译成功后：