Windows下用cmake编译Caffe

2018年8月24日 23:35:41

by ChrisZZ

系统： Windows10

编译器：Visual Studio 2015

构建器：CMake 3.12

Caffe: BVLC官方Caffe的windows分支

CUDA： 9.2

CuDNN：7.0

终端： cmd窗口

进cmd敲：

d:
cd work
git clone https://github.com/BVLC/caffe caffe-BVLC
cd caffe-BVLC
git checkout -b window origin/windows

2.1 cmake/Cuda.cmake，第7行，去掉20和21的算力支持（否则新版cuda会报错），改成：

set(Caffe_known_gpu_archs "30 35 50 52 60 61")

2.2 cmake/Cuda.cmake，第40行左右，手动设置nvcc架构版本，例如我是1080Ti显卡windows cmake 使用，使用6.1:

set(__nvcc_out "6.1")

(为啥这里的手动获得计算能力版本有问题？大概是字符编码导致输出这些警告，警告信息影响了恰当结果的获得)

2.3 下载依赖项

为了规避cmake下载遇到网络问题，以及设置配置项的方便，根据scripts/download_prebuilt_dependencies.py手动下载依赖文件：

WIN_DEPENDENCIES_URLS = {
    ('v120', '2.7'):("https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v120_x64_py27_1.1.0.tar.bz2",
                  "ba833d86d19b162a04d68b09b06df5e0dad947d4"),
    ('v140', '2.7'):("https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v140_x64_py27_1.1.0.tar.bz2",
                  "17eecb095bd3b0774a87a38624a77ce35e497cd2"),
    ('v140', '3.5'):("https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v140_x64_py35_1.1.0.tar.bz2",
                  "f060403fd1a7448d866d27c0e5b7dced39c0a607"),
}

下载后的压缩包放到C:\Users\chris.caffe\dependencies\download\目录下，然后解压到download的同级目录

接下来，如果你要用cuda编译，那么一定要修改C:\Users\chris.caffe\dependencies\libraries_v140_x64_py27_1.1.0\libraries\include\boost-1_61\boost\config\compiler\nvcc.hpp：注释掉最后三行：

#if !defined(__CUDACC_VER__) || (__CUDACC_VER__ < 70500)
#   define BOOST_NO_CXX11_VARIADIC_TEMPLATES
#endif

（原因：cuda7.5以后，__CUDACC_VER__被取消，应当使用__CUDACC_VER_MAJOR_和__CUDACC_VER_MINOR__等windows cmake 使用，参考、）。

2.4 开始编译

回到刚刚的cmd，继续敲：

set WITH_NINJA=0  # 不使用ninja，用visual studio。因为编译中间出了问题在vs里面好查看。
# set CPU_ONLY=1  # 如果没有GPU，或者就是想编译CPU版本，则开启
.\scripts\build_win.cmd # 前面我写明的修改项都完成后，本行会顺利执行。

编译之后出现“锟斤拷”的乱码可以忽略，最后：

本文来自电脑杂谈，转载请注明本文网址：
http://www.pc-fly.com/a/ruanjian/article-121285-1.html