记录编译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之旅。