博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Ubuntu14.04 安装CUDA7.5 + Caffe + cuDNN
阅读量:5922 次
发布时间:2019-06-19

本文共 2948 字,大约阅读时间需要 9 分钟。

本系列文章由
@yhl_leo
出品,转载请注明出处。
文章链接:


花了一天时间,在电脑上安装配置了Caffe深度学习框架,网上的很多教程和指导都已经过期,中间辗转耗费了点时间,这里把个人认为最简单的方法整理如下。

1 版本

  • 笔记本:ThinkPad W541
  • 双显卡:
    • Intel(R) HD Graphics Family
    • NVIDIA Quadro K2100M

2 准备工作

系统原本是想安装版本Wily,之前也一直在使用,但是无奈,电脑无法安装该版本,就选用了14.04。

安装一些基本依赖项:

$ 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

安装NVIDIA显卡驱动,我是先卸载已有版本,再安装,卸载方法:

sudo apt-get remove --purge nvidia*

重启后登录时,可能会循环出现填写登录密码,不能进入系统的情况,按Ctrl+Alt+F1,登录,安装显卡驱动:

$ sudo apt-get install nvidia-current

或者指定版本:

$ sudo add-apt-repository ppa:xorg-edgers/ppa$ sudo apt-get update$ sudo apt-get install nvidia-352

电脑是集成显卡(NVIDIA独显的不需要这么做),网友建议在安装前禁掉一些服务,以保证成功。我在安装的时候,只禁掉一项:

$ sudo service lightdm stop

(系统启动后,不要登录进入桌面,直接按Ctrl+Alt+F1,如果你玩过云服务器,应该很熟悉,接着输入系统用户名和登录密码,输入上述的指令,回到图形界面的方法是按Ctrl+Alt+F7,记住这个操作,14.04版本出现输入密码不能进入桌面的情况比较常见,我安装搜狗拼音都出现了……)

3 CUDA

网上很多安装6.x,7.0版本的方法非常繁琐,个人选用了相对简单的安装方法。

首先在官网上下载安装文件(链接前文已经提供):

cuda

我下载的是deb本地安装文件,下载完成后,按照文档提示的命令安装:

$ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb$ sudo apt-get update$ sudo apt-get install cuda

下载了下方的Installation Guide for Linux,里面有关于环境变量的设置方法:

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

其他安装文件的配置方式,文档里也都有。

4 Caffe

下载caffe:

$ git clone https://github.com/BVLC/caffe.git

如果你没安装Git,请阅读博客:。

然后编译caffe:

$ cd ~/caffe$ sudo cp Makefile.config.example Makefile.config$ make all

有些教程也配置OpenCV,方法可以阅读博客: 。

配置OpenCV,请阅读:(原贴已删)

配置OpenCV,请阅读:。

5 cuDNN

cuDNN是GPU加速计算深层神经网络的库(下载链接,前文已提供)。

cuDNN

本人的下载文件是:cudnn-7.0-linux-x64-v4.0-prod.tgz

在终端中切换到文件所在文件夹,输入下面指令:

$ sudo tar xvf cudnn-7.0-linux-x64-v4.0-prod.tgz$ cd cuda/include$ sudo cp *.h /usr/local/include/$ cd ../lib64$ sudo cp lib* /usr/local/lib/$ cd /usr/local/lib$ sudo chmod +r libcudnn.so.4.0.4$ sudo ln -sf libcudnn.so.4.0.4 libcudnn.so.4$ sudo ln -sf libcudnn.so.4 libcudnn.so$ sudo ldconfig

(我安装的是libcudnn.so.4.0.7,跟上面的例子对应就好)

然后切换到caffe根目录下,将Makefile.config中的USE_CUDNN行前的#去掉:

$ cd ~/caffe$ sudo vi Makefile.config

config

保存后重新编译:

$ sudo make clean$ sudo make all$ sudo make test  $ sudo make runtest

本人编译的时候最后一步的时候,出现这样的错误:

libcudart.so.7.5 cannot open shared object file: No such file or directory

网上有一堆介绍,怎么设置环境变量的(我们分明已经设置过了),可以这样检查是否已经添加环境变量:

$ echo $PATH$ echo $LD_LIBRARY_PATH

path

可以看到,环境变量已经添加好。

解决方法是这样,将一些文件复制到/usr/local/lib文件夹下:

$ sudo cp /usr/local/cuda-7.5/lib64/libcudart.so.7.5 /usr/local/lib/libcudart.so.7.5 && sudo ldconfig$ sudo cp /usr/local/cuda-7.5/lib64/libcublas.so.7.5 /usr/local/lib/libcublas.so.7.5 && sudo ldconfig$ sudo cp /usr/local/cuda-7.5/lib64/libcurand.so.7.5 /usr/local/lib/libcurand.so.7.5 && sudo ldconfig

再次,尝试sudo make runtest命令,出现如下:

maketest

至此,caffe安装完成。


源自:个人博客网站:

转载于:https://www.cnblogs.com/hehehaha/p/6332143.html

你可能感兴趣的文章
iOS-开发者相关的几种证书(转)
查看>>
sql 中的 STUFF()使用说明,以及千分位的常用函数
查看>>
one()方法的介绍
查看>>
斐波纳契数列非递归(C#)
查看>>
PyCharm安装及调试
查看>>
Android自定义XML属性
查看>>
C#判断WebService接口是否可用
查看>>
分享一个Eclipse代码配色教程与颜色主题插件
查看>>
deque简单解析
查看>>
提高我们微博互粉的效率,使用"一键关注"Chrome扩展程序
查看>>
Windows64位安装CPU版TensorFlow
查看>>
javascript reduce
查看>>
String 类的实现(2)引用计数与写时拷贝
查看>>
ubuntu 下 apache+tomcat整合_(mod-jk方法)[转]
查看>>
Flex数据交互之Remoting[转]
查看>>
转 Android之项目推荐使用的第三方库,有助于快速开发,欢迎各位网友补充
查看>>
commands - `ls`
查看>>
html 5 video 和audio属性
查看>>
编译器自动生成默认构造函数的四种情况
查看>>
LeetCode-反转整数
查看>>