新年编译。
新年伊始发现mkl、cuda、cudnn都有了新版本,于是在笔记本上重新编译了一下caffe。
首先还是官网下载最新版本的mkl、cuda、cudnn对应的windows平台版本并安装。
cudnn要解压到cuda工具的安装目录方便使用(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1)。
caffe checkout到windows branch。
编译环境我一直用的是vs2015,另外装一个miniconda(python环境)。这次miniconda安装也碰到个坑,安装时一定要选“只有我”,不然conda install需要管理员权限。另外最好选择添加系统路径,不然总是找不到conda.exe。
安装好miniconda后,就可以用anaconda的conda install安装ninja和cmake,用来编译caffe。
编译要在caffe文件夹的cmd下运行scripts\build_win.cmd,注意要修改这个文件把需要的mkl以及cuda配好。
:: Change the settings here to match your setup
:: Change MSVC_VERSION to 12 to use VS 2013, 14 to use VS 2015
if NOT DEFINED MSVC_VERSION set MSVC_VERSION=14
:: Change to 1 to use Ninja generator (builds much faster)
if NOT DEFINED WITH_NINJA set WITH_NINJA=1
:: Change to 1 to build caffe without CUDA support
if NOT DEFINED CPU_ONLY set CPU_ONLY=0
:: Change to generate CUDA code for one of the follow