Caffe需要预先安装一些依赖项,首先是CUDA驱动。不论是CentOS还是Ubuntu都预装了开源的nouveau显卡驱动(SUSE没有这种问题),如果不禁用,则CUDA驱动不能正确安装。以Ubuntu为例,介绍一下这里的处理方法,当然也有其他处理方法。
# sudo vi/etc/modprobe.d/blacklist.conf# 增加一行 :blacklist nouveausudoapt-get --purge remove xserver-xorg-video-nouveau #把官方驱动彻底卸载:sudoapt-get --purge remove nvidia-* #清除之前安装的任何NVIDIA驱动sudo service lightdm stop #进命令行,关闭Xserversudo kill all Xorg然后并安装 NVIDIA CUDA驱动包,接着安装安装BLAS、OpenCV、Boost这三个库。BLAS数学库可以是ATLAS, MKL, 或 OpenBLAS,OpenCV要求2.4以上版本,Boost要求1.55版本以上。可选安装Python MATLAB Caffe 库,还有 numpy , pandas 之类的Python类库。安装MATLAB,以确保mex 在path路径中,这些就够你折腾一个星期了,完事以后就可以编译安装了:
cp Makefile.config.example Makefile.config# Adjust Makefile.config (for example, if using Anaconda Python, or if cuDNN is desired)make allmake testmake runtest 哦对了,Caffe 显卡要求:Titan Xs, K80s, GTX 980s, K40s, K20s, Titans, and GTX 770s