码迷,mamicode.com
首页 > 系统相关 > 详细

ubuntu 14.04 cuda7.5 caffe安装

时间:2016-07-09 10:27:43      阅读:263      评论:0      收藏:0      [点我收藏+]

标签:

最近新接触深度学习就从入门开始吧:新安装cuda,caffe安装流程很简单,网上到处都有

1:安装cuda前要禁用nouveau驱动

    按Ctrl+Alt+F1进入命令提示符,新建一个黑名单文件

# sudo vi /etc/modprobe.d/blacklist-nouveau.conf

输入

blacklist nouveau
options nouveau modset=0

保存退出(:wq)

然后执行

# sudo update-initramfs -u

执行 lspci | grep nouveau查看是否有内容

# lspci | grep nouveau

如果没有内容 ,说明禁用成功,如果有内容,就重启一下再查看

# sudo reboot

重启后,进入登录界面的时候,不要登录进入桌面,直接按Ctrl+Alt+F1进入命令提示符。

 

2:安装cuda

https://developer.nvidia.com/cuda-downloads下载对应版本的deb文件,

dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb

sudo apt-get update

sudo apt-get install cuda

配置环境变量等

输入:
sudo nano /etc/profile
- 在文件的最后一行添加 PATH="$PATH:/usr/local/cuda-7.5/bin"
- 然后保存并退出
输入:
source /etc/profile
输入:
echo $PATH
输入:
sudo nano /etc/ld.so.conf
- 在第二行写入 include /usr/local/cuda-7.5/lib64
注:如果是第一次装机,先要设置一下 root 密码
输入:
sudo passwd root
root 密码设置好之后,输入 su 进入 root 账号
输入: ldconfig
输入: exit

3:安装caffe

先下载caffe

# sudo git clone https://github.com/BVLC/caffe.git

然后安装一堆第三方库

 
# sudo apt-get install libatlas-base-dev
# sudo apt-get install libprotobuf-dev
# sudo apt-get install libleveldb-dev
# sudo apt-get install libsnappy-dev
# sudo apt-get install libopencv-dev
# sudo apt-get install libboost-all-dev
# sudo apt-get install libhdf5-serial-dev
# sudo apt-get install libgflags-dev
# sudo apt-get install libgoogle-glog-dev
# sudo apt-get install liblmdb-dev
# sudo apt-get install protobuf-compiler

接下来,编译caffe


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

至此,caffe安装完成。


配置运行环境


caffe运行时需要调用cuda的库,我们在/etc/ld.so.conf.d目录下新建一个cafe.conf文件,将所需要用的库的目录写入


# sudo vi /etc/ld.so.conf.d/caffe.conf

添加内容:


/usr/local/cuda/lib64

保存退出(:wq)


更新配置


# sudo ldconfig
 

ubuntu 14.04 cuda7.5 caffe安装

标签:

原文地址:http://www.cnblogs.com/fanghong/p/caffe-0.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!