安装Centos6.5系统

紧接着找到root ALL=(ALL) ALL这一行,在下一行添加lei ALL=(ALL) ALL,保存退出编辑,:wq即可

基本依赖库的安装

安装caffe

wget -c -O caffe.zip https://github.com/BVLC/caffe/archive/master.zip
$ cp Makefile.config.example Makefile.config
# Adjust Makefile.config (for example, if using Anaconda Python)
$ make all -j 32
$ make test
$ make runtest

安装显卡驱动,配置CAFFE GPU模式

  1. 下载驱动,网址为 http://www.nvidia.cn/Download/index.aspx?lang=cn
  2. 安装内核源码,以提供驱动编译</br> 上此网址http://www.rpmfind.net/linux/rpm2html/search.php?下载centos6.5对应的内核源码,kernel-devel-2.6.32-431.29.2.el6.x86_64.rpm、kernel-headers-2.6.32-431.29.2.el6.x86_64.rpm
  3. yum -y install kernel-devel-2.6.32-431.29.2.el6.x86_64.rpm kernel-headers-2.6.32-431.29.2.el6.x86_64.rpm
  4. 修改/etc/modprobe.d/blacklist.conf 文件,以阻止 nouveau 模块的加载 方法: 在文件末尾添加blacklist nouveau
  5. 重新建立initramfs image文件,该文件主要功能为维护系统内核,提供系统内核加载的位置信息等
    mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
    dracut /boot/initramfs-$(uname -r).img $(uname -r)
    
  6. 修改/etc/inittab,使系统开机进入init 3文本模式: 将最后一行“id:5:initdefault:”修改成“id:3:initdefault:” 说明:5代表系统启动时默认进入x-window图形界面,3代表默认进入终端模式。
  7. 阻止kernel加载nouveau模块
    sudo vim /boot/grub/grub.conf
    #在当前内核版本选项的kernel行的末尾加上 rdblacklist=nouveau vga=792
    #重启 
    reboot
    
  8. ​安装NVIDIA驱动
    1.输入root和password,进入根用户模式下,确保nouveau kernel driver没有被加载
    # lsmod | grep nouveau
    2.进入驱动程序所在目录,开始安装
    # sudo chmod +x NVIDIA-Linux-x86-331.20.run
    # sudo ./NVIDIA-Linux-x86-331.20.run
    3.安装过程中,根据提示选择accept,yes 或 OK,即可完成安装:
      如果提示有旧驱动,询问是否删除旧驱动,选Yes;
      如果提示缺少某某模块(modules),询问是否上网下载,选no;
      如果提示编译模块,询问是否进行编译,选ok;
      如果提示将要修改xorg.conf,询问是否允许,选Yes;
      接下来就是等待安装完成。
    
  9. 修改/etc/inittab,使系统开机进入init 5图形界面模式
    将最后一行“id:3:initdefault:”修改成“id:5:initdefault:”
    重启
    #reboot
    
  10. 检查nvidia驱动是否安装成功
    1.对于一般的普通显卡,geforce系列等
    登陆后,在系统- 首选项里可以看到NVIDIA X Server Settings菜单,即可证明安装成功
    2.对于Tesla系列等高端计算卡
    可以通过命令nvidia-smi来查看GPU的配置运行情况
    

安装cuda

  1. 下载CUDA 6.5的RUN文件
    $ chmod +x cuda_6.5_XXXX.run
    $ ./cuda_6.5_XXXX.run
    

    跳过第一项的驱动安装,后面要么yes要么默认就可以了。

  2. 环境变量配置
    /usr/local/cuda/bin加入PATH
    /usr/local/cuda/lib64加入/etc/ld.so.conf
    
  3. 重新编译安装caffe
    回到安装caffe的流程上
    注释 CPU_ONLY := 1,make clean,make all -j 32,make install