caffe-Anaconda-python3.5

如果你是Ubuntu16.04系统

那么会有python2.7和python3.5

但是我们希望通过anaconda来编译caffe

其他的和caffe编译一样:

sudo apt-get install -y update
sudo apt-get install -y upgrade

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



sudo apt-get install -y git
sudo apt-get install -y cmake
sudo apt-get install -y build-essential
sudo apt-get install -y python-pip

sudo apt-get install -y gfortran
sudo apt-get install python-dev

sudo apt-get install -y python-matplotlib


sudo apt-get install -y python-skimage
sudo apt-get install -y graphviz

更改caffe中的一下配置:

Makefile:

#PYTHON_LIBRARIES ?= boost_python python2.7
PYTHON_LIBRARIES ?= boost_python-py35 python3.5m

Makefile.config:

我们新建了py3.5的环境:

pytorch2caffe-p35

USE_CUDNN := 1


CUDA_DIR := /usr/local/cuda

CUDA_ARCH := -gencode arch=compute_50,code=sm_50 \
             -gencode arch=compute_52,code=sm_52 \
             -gencode arch=compute_61,code=sm_61 \
	     -gencode arch=compute_61,code=compute_61


BLAS := atlas


ANACONDA_HOME := /home/boyun/anaconda3/envs/pytorch2caffe-p35

PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
		$(ANACONDA_HOME)/include/python3.5m \
		$(ANACONDA_HOME)/lib/python3.5/site-packages/numpy/core/include \

PYTHON_LIB := $(ANACONDA_HOME)/lib

WITH_PYTHON_LAYER := 1

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial

然后打开anaconda,进入我们的环境,cd到caffe的python文件夹下:

cd ~/caffe/python
for req in $(cat requirements.txt); do pip install -i https://pypi.tuna.tsinghua.edu.cn/simple $req; done

 或者直接安装如下:

sudo pip install pip==9.0.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

sudo pip install protobuf==2.6.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install ipython==3.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

sudo pip install pydot
sudo pip install Cython -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install scipy -i https://pypi.tuna.tsinghua.edu.cn/simple

sudo pip install scikit-image

sudo pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install ipython -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install h5py -i https://pypi.tuna.tsinghua.edu.cn/simple

sudo pip install leveldb -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install networkx -i https://pypi.tuna.tsinghua.edu.cn/simple

sudo pip install nose -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install pandas -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install python-dateutil -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install protobuf -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install python-gflags -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install pyyaml -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install Pillow -i https://pypi.tuna.tsinghua.edu.cn/simple
sudo pip install six -i https://pypi.tuna.tsinghua.edu.cn/simple

然后,加入sudo gedit ~/.bashrc

source ~/.bashrc

猜你喜欢

转载自blog.csdn.net/baidu_40840693/article/details/91986739