标签:
最近新接触深度学习就从入门开始吧:新安装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
标签:
原文地址:http://www.cnblogs.com/fanghong/p/caffe-0.html