本人是技术小白,刚开始接触caffe。第一次写博客,目的就是为了把自己的笔记存档,方便以后查阅。所以如果有写的不好或者不对的地方,还望各位大佬多多指教。
caffe的安装是在ubuntu16.04下安装的,关于ubuntu系统的安装这里不再陈述,网上有大量的教程可供参考。这篇文章主要讲一下自己在安装配置caffe的过程中遇到的各种问题和各种坑。。。
1.安装相关依赖项
$sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
$sudo apt-get install --no-install-recommends libboost-all-dev
$sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
$sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
2.安装NVIDIA驱动
(1)查询NVIDIA驱动
首先去官网(http://www.nvidia.com/Download/index.aspx?lang=en-us)查看适合自己显卡的驱动,下面是适合我的显卡的驱动:
(2)安装驱动
安装之前先卸载已经存在的驱动版本:
这里有两种方法:
(1)利用“软件和更新”来安装,依次选择 系统设置->软件和更新->附加驱动->选择最新的驱动->应用更改
安装时可能遇到的问题:点击完应用更改一段时间后并没有成功安装,再次点击却出现闪退的现象,后来发现是因为依赖的问题,通过在终端输入以下命令:$sudo apt-get install -f
$sudo apt-get update
后 再次安装问题就解决了。
备注:之前都是用这种方法,可是后来重装系统后再次安装闪退,使用这个方法也解决不了,就用的命令行安装的
(2)命令行安装:
sudo add-apt-repository ppa:xorg-edgers/ppa
sudo apt-get update
sudo apt-get install nvidia-375 #注意在这里指定自己的驱动版本!
安装完成之后输入以下指令进行验证:
$sudo nvidia-smi
或者$sudo nvidia-settings
若列出了GPU的信息列表则表示驱动安装成功。
3.cuda 安装
这里我需要说一下,cuda7.5和cuda8.0我都安装过,开始安装cuda7.5的过程中,后来make时出现nvcc错误,查阅发现ubuntu16.04适合安装cuda8.0,果断转8.0
cuda都是在官网下载的,这里不再赘述
在终端cd到所下载的安装包所在的目录,进行MD5检验,确保安装包正常:
$sudo md5sum cuda.run
检验正确后,输入如下代码安装(第一行修改文件权限,第二行执行文件安装):
$ chmod 777 cuda.run
$ sudo ./cuda.run --override
注意后面的override是必须的,这样才能保证安装的过程中,不会出现编译器不支持的错误。另外,在选择条件的过程中,一定不要再次安装nvidia驱动,虽然cuda.run文件本身是包含又nvidia驱动的,但是本处直接安装会出错。下图是安装.run文件的配置:
这里可能会出现miss library的情况,解决方案参考http://stackoverflow.com/questions/22360771/missing-recommended-library-libglu-so
4.cudnn安装
下载地址:https://developer.nvidia.com/cudnn
cudnn-7.0-linux-x64-v4.0-prod.tgz
进入下载目录解压:
$ tar zxvf cudnn-7.0-linux-x64-v4.0-prod.tgz
cd进入cudnn解压之后的include目录,在命令行进行如下操作:
$ sudo cp cudnn.h /usr/local/cuda/include/
#复制头文件
再将lib64目录下的动态文件进行复制和链接:
$ sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
$ cd /usr/local/cuda/lib64/
$ sudo ls -al | grep libcudnn#查看已有链接
结果如下:
$ sudo rm -rf libcudnn.so libcudnn.so.4 #删除原有动态文件
$ ls -al | grep libcudnn #再次查看已有链接
建立新的链接
$ sudo ln -s libcudnn.so.4.0.7 libcudnn.so.4
$ sudo ln -s libcudnn.so.4 libcudnn.so
再次查看
$ ls -al | grep libcudnn
已链接好!
然后设置环境变量和动态链接库:
$ sudo gedit /etc/profile
然后再打开的文件末尾加上(“=”前后不要有空格)
export PATH=/usr/local