camke编译caffe
vs2015+cuda8.0 cudnn5 python2.7/
1、解压出原始的caffe-windows
2、新建一个build文件夹
3、放进去压缩包
4、修改common.cpp
如果想看训练日志怎么办,想看训练过程的accuracy和loss到底是多少怎么办?这就得自己动手了,改代码。
用vs2013打开在src文件夹里的common.cpp文件,添加头文件 #include<direct.h>
找到voidGloballnit()函数,在::google::InitGoogleLogging(*(pargv)[0]);下面一行添加如下代码:
_mkdir("./log/");
FLAGS_colorlogtostderr =true;//设置输出到屏幕的日志显示相应颜色
google::SetLogDestination(google::GLOG_FATAL,"./log/log_error_");
google::SetLogDestination(google::GLOG_ERROR,"./log/log_error_");
google::SetLogDestination(google::GLOG_WARNING,"./log/log_error_");
google::SetLogDestination(google::GLOG_INFO,"./log/log_info_");
FLAGS_max_log_size = 1024;//最大日志大小为 1024MB
FLAGS_stop_logging_if_full_disk =true;//磁盘写满时,停止日志输出
然后编译common.cpp文件,然后重新编译生成caffe.exe文件,这样最后训练调用caffe.exe后就会在run.bat同级目录下生成一个log文件,里面就是训练日志,训练的整个过程都在里面,你可以提取里面的数据,或画loss和accuracy关于迭代次数的曲线图,这样更直观,这对调试网络很有帮助。
5、cmake编译
会出现这种错误,没关系。然后修改,重新编译
看一下python、matlab是否找到,就成功了。
祝好运!