1 ubuntu系统安装
ubuntu 14.04下载地址:http://www.releases.ubuntu.com/14.04/
可以直接点击以下地址直接下载
http://www.releases.ubuntu.com/14.04/ubuntu-14.04.5-desktop-amd64.iso
2 caffe环境安装
General dependencies
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler sudo apt-get install --no-install-recommends libboost-all-dev
BLAS: sudo apt-get install libatlas-base-dev
Python: sudo apt-get install python-dev python-pip
ubuntu 14.04 :sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
CUDA: 使用的cuda7.5版本
sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb sudo apt-get update sudo apt-get install -y cuda设置cuda环境:
在/etc/profile文件最后添加CUDA环境变量:
sudo gedit /etc/profile
PATH=/usr/local/cuda/bin:$PATH
export PATH
保存后, 执行下列命令, 使环境变量立即生效
source /etc/profile
同时需要添加lib库路径: 在 /etc/ld.so.conf.d/加入文件 cuda.conf
sudo gedit /etc/ld.so.conf.d/cuda.conf
内容如下:
/usr/local/cuda/lib64
保存后,执行下列命令使之立刻生效
sudo ldconfig
CUDNN:使用4.0版本
tar -zxvf cudnn-7.0-linux-x64-v4.0-prod.tgz cd cuda sudo cp lib64/lib* /usr/local/cuda/lib64/ sudo cp include/cudnn.h /usr/local/cuda/include/ cd /usr/local/cuda/lib64/ sudo chmod +r libcudnn.so.4.0.7 sudo ln -sf libcudnn.so.4.0.7 libcudnn.so.4 sudo ln -sf libcudnn.so.4 libcudnn.so sudo ldconfig
CUDNN:使用5.0版本
tar -zxvf cudnn-7.5-linux-x64-v5.0-ga.tgz cd cuda sudo cp lib64/lib* /usr/local/cuda/lib64/ sudo cp include/cudnn.h /usr/local/cuda/include/ cd /usr/local/cuda/lib64/ sudo chmod +r libcudnn.so.5.0.5 sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5 sudo ln -sf libcudnn.so.5 libcudnn.so sudo ldconfig
3 python 接口编译
安装git和下载caffe源码:
sudo apt-get install build-essential cmake git pkg-config git clone https://github.com/BVLC/caffe.git
下载python依赖环境
sudo su sudo apt-get install gfortran pip install --upgrade pip for req in $(cat requirements.txt); do pip install $req; done // 推荐使用国内镜像安装,你懂的,速度快 for req in $(cat requirements.txt); do pip install $req -i http://pypi.douban.com/simple; donefor req in $(cat requirements.txt); do pip install $req -i https://pypi.tuna.tsinghua.edu.cn/simple; done
编译:先拷贝一份Makefile.config文件,取消一下注释
cp Makefile.config.example Makefile.config
# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1# Uncomment to support layers written in Python (will link against Python libs)
WITH_PYTHON_LAYER := 1使用如下命令编译caffe和python接口
make -j
make py编译 pycaffe时报错:fatal error: numpy/arrayobject.h没有那个文件或目录
其实numpy已经是安装的,anaconda2里面有,Python中import numpy也没有问题,但就是在此处报错,解决方法:
sudo apt-get install python-numpy
4 使用spyder调用python接口和kdevelop调用c++
安装
sudo pip install spyder sudo apt-get install kdevelop
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:ubuntu14.04安装caffe+python接口 - Python技术站