前言
为了顺利的把Caffe跑起来,昨天把公司电脑系统换成了Ubuntu14.04。其实编译Caffe很简单,麻烦的地方就在于安装各种库,尤其是cuda toolkit。百度了一下cuda在ubuntu下面的安装教程,真的是五花八门,最后找了一个很靠谱的教程。
安装步骤
第一步 下载cuda toolkit安装包
去CUDA官网下载cuda toolkit的安装包
根据自己的系统选择合适的安装包,此处我选的是 Ubuntu14.04 Local Package Installer,下载完成后会看到一个名为cuda-repo-ubuntu1404_7.0-28_amd64.deb 的文件。
第二步 安装cuda toolkit
可以选择在命令行安装deb文件,也可以直接双击进行安装。命令行安装命令如下:
$ sudo dpkg -i cuda-repo-ubuntu1404_7.0-28_amd64.deb
$ sudo apt-get update
然后就可以使用apt-get install来安装cuda toolkit了
第三步 设置环境变量
设置环境变量的方式也很多,这里只介绍其中的一种
打开文件.profile(只对当前用户生效):
$sudo gedit ~/.profile
在文件的末尾添加如下几行:
export CUDA_HOME=/usr/local/cuda-7.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
PATH=${CUDA_HOME}/bin:${PATH}
export PATH
保存文件之后在命令行执行一下,让配置内容生效
$sudo source ~/.profile
到这个地方,cuda已经配置好了。为了验证一下cuda能不能使用,可以编译一下官方自带的samples
$cd /usr/local/cuda/samples/
$sudo make
例子编译好之后,在samples下面的子文件bin/x86_64/linux/release中,运行编译好的可执行程序
$cd bin/x86_64/linux/release
$./deviceQuery
如果一切正常,那么该程序会打印出本机cuda device的一些信息