ubuntu14.04下caffe框架的安装

来源:互联网 发布:网络应用模型 编辑:程序博客网 时间:2024/05/17 11:56

ubuntu14.04下caffe框架的安装

  • ubuntu1404下caffe框架的安装
    • OS的安装
    • 驱动与环境的配置
      • 开始配置环境
      • 安装CUDA
      • 安装BLAS OpenCV等配件
      • 安装caffe

OS的安装

从网上下载ubuntu14.04,最好不要用16.0+的版本,对coffe的兼容性不好。下载镜像时,可到各高校的镜像站下载,如中大镜像或ustc开源镜像,基本可保证安全无毒。

驱动与环境的配置

由于新安装的ubuntu里标配的是vi的tiny版本,用起来很不习惯,所以这里我先把它卸掉了,再重装一个完全版的,这样用起来就舒服多了。

sudo apt-get remove vim-commonsudo apt-get install vim

开始配置环境

说是配置环境,其实就是换个显卡驱动。因为不论是CentOS还是Ubuntu都预装了开源的nouveau显卡驱动(SUSE没有这种问题),如果不禁用,则CUDA驱动不能正确安装。(此处参考网上教程)

# 打开配置文件# 在里面加入blacklist nouveau# blacklist的作用就是禁用表sudo vi/etc/modprobe.d/blacklist.conf

卸载已有显卡驱动

sudo apt-get --purge remove xserver-xorg-video-nouveau   #把官方驱动彻底卸载:sudo apt-get --purge remove nvidia-*    #清除之前安装的任何NVIDIA驱动sudo service lightdm stop    #进命令行,关闭Xserversudo kill all Xorg

在卸载驱动的时候,电脑会黑屏,这时就要用alt+ctrl+F4进入命令行界面继续操作了

安装CUDA

此部分参考http://blog.csdn.net/xizero00/article/details/43227019

首先下载nvidia cuda的仓库安装包(我的是ubuntu 14.04 64位,所以下载的是ubuntu14.04的安装包,如果你是32位的可以参看具体地址)

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_6.5-14_amd64.deb

下载完成之后可以使用如下命令安装它

sudo dpkg -i cuda-repo-ubuntu1404_6.5-14_amd64.deb

安装好仓库之后,就可以更新你的本地仓库

sudo apt-get update

最后开始安装cuda以及显卡驱动(安装cuda的同时就会把显卡驱动也全部安装好。但是下载的时间有点长。)

sudo apt-get install cuda

安装完之后你需要设置环境变量:

export PATH=/usr/local/cuda-6.5/bin:$PATH  export LD_LIBRARY_PATH=/usr/local/cuda-6.5/lib64:$LD_LIBRARY_PATH  

设置完毕之后,你还可以选择是否安装cuda附带的示例代码:

cuda-install-samples-6.5.sh <dir> 

注意事项

使用这个方法的时候千万要注意几个问题:
(1)cuda6.5已经不支持老旧的显卡了所以sm11 等等都必须删除。可以参考我的另一个文章,关于编译opencv3.0的
(2)ubuntu14.04是64位的,并且不要一开始就更新系统补丁什么的,因为系统更新过之后,再安装显卡驱动就会无法进入图形界面,我查看了相关的日志发现是卡在了dbus那边。所以,我建议一安装好ubuntu 14.04就不要更新系统补丁。

安装BLAS OpenCV等配件

sudo apt-get install libatlas-base-devsudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-devsudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compilersudo apt-get install python-pip sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose  

安装caffe

从github上获取caffe,复制配置文件,执行make指令

git clone https://github.com/BVLC/caffe.gitcd caffecp Makefile.config.example Makefile.configmake allmake  pycaffe
0 0
原创粉丝点击