记一次Caffe的安装

运河杯需要所以要安装Caffe框架用于图像识别,记录一下遇到的坑

基本都是按照Caffe的github上面的wiki安装下来的,挺顺利的除了Cudnn。 用的Ubuntu16.04 Lts 显卡是GeForce 820M(N多年前的卡了,机器学习真的贵)和一张集显.

Cuda

cuda直接从NVIDIA下载对应的deb包(我用的在线安装板),这里面的包好像包括了N卡的驱动. 第一遍装的时候cuda直接把原来的Nvidia驱动覆盖了,装的Cuda8.0貌似自带的375驱动. 出现了的问题是当切换到N卡的时候开机会显示Low graphic mode,要么是开机循环登录

解决办法

对于Low graphic Mode,按下Ctrl+Alt+F1~F6 登录一下,使用 shell sudo prime-select intel 这个只是切换到了N卡,每次关机之前还是要记得切回Intel(蛋疼无比) 最终解决办法是卸载掉全部的N卡驱动 Cuda 然后重装cuda(不挂代理,国内访问下载也挺快的)

opencv

我安装的时候挺顺利的没什么大问题

cudnn

cudnn可以大幅度加速计算速度(那么棒,那当然上上上),无论我换Cudnn6.0还是5.1在caffe开始计算的时候都显示我的Cudnn出了问题 text Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0) CUDNN_STATUS_ARCH_MISMATCH 查了一下是我的显卡问题不支持cudnn, 因为笔记本上的显卡的CUDA Capability是2.1,而官方的cudnn加速是不支持3.0以下的版本的，因此只能在Makefile.config中注释掉USE_CUDNN这行,重新make一下caffe

caffe

没遇到问题。