caffe linux 教程,Caffe 深度学习入门教程 - 安装配置Ubuntu14.04+CUDA7.5+Caffe+cuDNN_Linux教程_Linux公社-Linux系统门户网站...
安裝配置Ubuntu14.04+CUDA7.5+Caffe+cuDNN
一、版本
Linux系統:Ubuntu 14.04 (64位)
顯卡:Nvidia K20c
cuda: cuda_7.5.18_linux.run
cudnn: cudnn-7.0-linux-x64-v4.0-rc
二、下載
Ubuntu 14.04下載地址:http://www.ubuntu.com/download/desktop (64bit)
cuda7.5下載地址:https://developer.nvidia.com/cuda-downloads/ ?,下載對應的操作系統和版本cuda_7.5.18_linux.run,放到~根目錄下
cudnn下載址:https://developer.nvidia.com/cudnn, 需要注冊,并通過審核才能下載,下載相應文件cudnn-7.0-linux-x64-v4.0-rc.tgz,?放到~根目錄下.審核一般一至兩天,等不及的同學可到Linux公社1號FTP服務器下載
------------------------------------------分割線------------------------------------------
用戶名:ftp1.linuxidc.com
密碼:www.linuxidc.com
在 2016年LinuxIDC.com\11月\Caffe 深度學習入門教程\
------------------------------------------分割線------------------------------------------
三、開始安裝
1、安裝ubuntu,此文不關注。
2、禁用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進入命令提示符。
3、安裝cuda 7.5
先安裝一些常用的軟件,注意整個過程需要聯網
#sudo service lightdm stop#sudo apt-get install g++#sudo apt-get install git#sudo apt-get install freeglut3-dev
接下來進入~根據目錄安裝cuda 7.5
#cd#sudo sh cude_7.5.18_linux.run
安裝的時候,要讓你先看一堆文字(EULA),我們直接不停的按空格鍵到100%,然后輸入一堆accept,yes,yes或回車進行安裝。
安裝完成后,重啟,然后用ls查看一下,是否生成了四個左右以nvidia開頭的文件夾
#ls /dev/nvidia*
如果有,說明安裝成功了,如果沒有,可能不成功,需要卸載重裝。卸載命令如下:
#sudo /usr/local/cuda-7.5/bin/uninstall_cuda_7.5.pl#sudo /usr/bin/nvidia-uninstall
如果你還不放心是否安裝成功,請參考其它教程,編譯Samples進行測試。
最后,配置環境變量,我們直接放在系統配置文件profile里面,先打開profile文件
#sudo vi /etc/profile
在最后面加入兩行代碼,如果你還不會用vi進行編輯,請百度
export PATH=/usr/local/cuda-7.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
保存退出,至此cuda 7.5安裝完畢。
4、安裝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
接著,安裝opencv
#cd caffe#sudo git clone https://github.com/jayrambhia/Install-OpenCV#cd Install-OpenCV/Ubuntu#sudo sh dependencies.sh#cd 2.4#sudo sh opencv2_4_10.sh
接下來,編譯caffe
#cd ~/caffe#sudo cp Makefile.config.example Makefile.config#make all
至此,caffe安裝完成。
5、配置運行環境
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
6、測試caffe
下載mnist數據
#cd ~/caffe#sudo sh data/mnist/get_mnist.sh#sudo sh examples/mnist/create_mnist.sh
運行時,如果你有GPU,則不需要修改配置文件,如果沒有gpu,則需要修改配置文件lenet_solver.prototxt
#sudo vi examples/mnist/lenet_solver.prototxt
將最后一行的solver_mode:GPU改為solver_mode:CPU
配置好后,就可以運行了
#sudo sh examples/mnist/train_lenet.sh
注意,運行caffe程序時,必須在caffe的根目錄下,不然會出錯
7、安裝cudnn進行加速
假設事先已經下載好cudnn-7.0-linux-x64-v4.0-rc.tgz放到~根目錄
#cd#sudo tar xvf cudnn-7.0-linux-x64-v4.0-rc.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
將caffek根目錄下 Makefile.config中 USE_CUDNN 行的注釋去除,然后重新進行編譯。
#cd ~/caffe#sudo vi Makefile.config
將USE_CUDNN 行的注釋符號#去除,即 USE_CUDNN := 1
然后重新編譯
#sudo make clean#sudo make all
到此,全部安裝完成。
如果沒有gpu,則不能用cudnn。
總結
以上是生活随笔為你收集整理的caffe linux 教程,Caffe 深度学习入门教程 - 安装配置Ubuntu14.04+CUDA7.5+Caffe+cuDNN_Linux教程_Linux公社-Linux系统门户网站...的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Python学习笔记:web开发1
- 下一篇: Python学习笔记:web开发2