ubunt16.04下caffe环境安装
来源:互联网 发布:全局优化算法举例 编辑:程序博客网 时间:2024/06/05 08:44
一. 系统初始环境
系统:Ubuntu16.04: ubuntu-16.04-desktop-amd64.iso
cuda安装文件: cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb.44-1_amd64-deb,下载链接点击, linux-x86架构-ubuntu-16.04-deb(local)
cudnn安装文件: cudnn-8.0-linux-x64-v5.0-ga.solitairetheme8,下载链接点击, 适用cuda8.0有5.1和5.0版,这里用5.0版,区别应该不大
caffe源代码: github链接 ,或者运行git clone https://github.com/BVLC/caffe.git
二. 安装cuda8.0
- sudo apt-get update
- sudo apt-get upgrade
- 删除自带显卡驱动 sudo apt-get –purge remove nvidia-*
- sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb.44-1_amd64-deb (这个文件包含了nvidia-367的驱动)
- 安装cuda: sudo apt-get install cuda
- reboot
测试安装结果:
1. cd /usr/local/cuda/samples/1_Utilities/deviceQuery
2. sudo make
3. ./deviceQuery
测试成功显示结果(部分):
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 8.0,
CUDA Runtime Version = 8.0, NumDevs = 1, Device0 = GeForce GTX 960M
Result = PASS
如果出现Result = Failed之类的表示cuda安装失败
添加环境变量:
在/etc/profile中添加
export PATH=/usr/local/cuda-8.0/bin:
三. 安装cudnn5.0
- 解压安装包
tar zxvf cudnn-8.0-linux-x64-v5.0-ga.solitairetheme8 - 复制文件
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn.so.5.0.5 /usr/local/cuda/lib64/ - 建立软链接
cd /usr/local/cuda/lib64/
sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so
四. 安装caffe
- 安装依赖库
sudo apt-get install build-essential
sudo apt-get install git cmake gedit doxygen
sudo apt-get install python-numpy python-pip cython easydict
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libopencv-dev (想使用opencv3的可以尝试跳过这一步自行安装opencv3)
sudo apt-get install –no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev - 下载源代码并解压
git clone https://github.com/BVLC/caffe.git
unzip caffe-master.zip 修改配置文件Make.config
cd caffe-master
cp Makefile.config.example Makefile.config
sudo gedit Makefile.config开启GPU模式:将#USE_CUDNN := 1前的#注释去掉,表示使用cudnn,如果不是使用GPU,可以将#CPU_ONLY := 1前得注释去掉pycaffe模块配置:修改\#Whatever else you find you need goes here.INCLUDE_DIRS:= $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS:=$(PYTHON_LIB) /usr/local/lib /usr/lib/usr/lib/x86_64-linux-gnu/hdf5/serial
如果要用opencv3编译的话还需要改动:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5 # handle IO dependencies USE_LEVELDB ?= 1 USE_LMDB ?= 1 USE_OPENCV ?= 1 ifeq ($(USE_LEVELDB), 1) LIBRARIES += leveldb snappy endif ifeq ($(USE_LMDB), 1) LIBRARIES += lmdb endif ifeq ($(USE_OPENCV), 1) LIBRARIES += opencv_core opencv_highgui opencv_imgproc ifeq ($(OPENCV_VERSION), 3) LIBRARIES += opencv_imgcodecs endif endif
4. 编译caffe
mkdir build
cd build
cmake ..
make all -j8
编译pycaffe
cd caffe-master
make pycaffe编译测试模块
make all
make test
make runtest
测试成功显示结果(部分):
[———-] Global test environment tear-down
[==========] 2091 tests from 283 test cases ran. (415487 ms total)
[ PASSED ] 2091 tests.
添加环境变量:
在~/.bashrc中添加
export PYTHONPATH=/home/hyzhan/caffe-master/:$PYTHONPATH
export PYTHONPATH=/home/hyzhan/caffe-master/caffe/python:$PYTHONPATH
- ubunt16.04下caffe环境安装
- Ubunt16.04安装CUDA cuDNN OpenCV Caffe完整教程
- ubunt16.04安装awscli
- Ubuntu14.04下安装caffe环境
- tk1环境下安装 caffe
- ubunt16 安装jdk7
- Ubunt16.04下编译Linux遇到的问题
- Ubuntu16.04环境下安装caffe及运行faster-rcnn
- Ubuntu16.04环境下安装CUDA、cudnn、Caffe、Tensorflow、pytorch
- win7下caffe环境安装过程
- CentOs7下,Caffe环境安装问题解决
- Win10环境下caffe安装与编译
- Ubuntu14.04下安装Caffe
- ubuntu16.04下安装caffe
- Ubuntu16.04下安装Caffe
- Ubuntu14.04下安装caffe环境之CUDA8.0的安装
- 【Caffe安装】Ubuntu14.04下安装Caffe总结
- Ubuntu16.04下环境配置Caffe
- 6.13
- 尚学堂百战程序员1573题 第五章
- Spring Framework之BeanFactoryPostProcessor和BeanPostProcessor(TODO)
- 时间复杂度的计算
- MyBatis中$与#的区别
- ubunt16.04下caffe环境安装
- ReportViewer 显示问题
- 7. Reverse Integer
- C语言中的常量
- Ubuntu 安装部署 Hadoop 2.6.5 实例演示
- Spring Boot使用Swagger2
- python以及pip在win10中的安装(64位操作系统)
- unity的生命周期
- 案例项目-DSTLD