Win10+VS2015+Anaconda3+Python3.5+Cuda10.0编译Caffe

记录编译GPU版Caffe的各种坑

准备工作

Cuda10.0

安装Cuda10.0及对应版本的cuDNN

CMake

下载并安装CMake，注意需要将其添加到环境变量中

Anaconda

我安装的是Anaconda2018.12，默认Python3.7，Caffe支持Python3.5，因此需要创建Caffe环境。

conda create -n caffe python=3.5

将Python路径添加到环境变量中，注意要把Python3.5的放在前面

否则在编译时会出现错误

CMake Error at cmake/WindowsDownloadPrebuiltDependencies.cmake:40 (message):
Could not find url for MSVC version = 1900 and Python version = 3.7.

Caffe

下载并解压caffe
https://github.com/BVLC/caffe/tree/windows

修改build_win.cmd

使用notepad++打开yourpath/caffe-windows/scripts/build_win.cmd做如下修改

运行build_win.cmd

这个过程中会将相关依赖项下载并解压到C:\Users\yourname.caffe\dependencies，此时会报如下错误

解决方法如下，我的显卡是1070Ti，Pascal架构

如果架构不匹配，在训练过程中会报cudaSuccess(48vs.0)错误

重新运行build_win.cmd，会报如下错误

按照红字提示，找到并打开nvcc.hpp，将22~24行注释掉

重新运行build_win.cmd开始漫长的编译过程，编译完成之后将path/caffe-windows\python下的caffe目录整体拷贝到D:\Anaconda3\envs\caffe\Lib\site-packages目录下即可愉快的开始Caffe之旅。