[ubuntu] Caffe 라이브러리 설치 방법

터미널창을 먼저 킨다. 

1. dependencies package 다운 받기

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler python-dev libgflags-dev libgoogle-glog-dev liblmdb-dev libboost-all-dev libatlas-base-dev libopenblas-dev libopencv-dev python-opencv python-pip python-dev python-protobuf build-essential
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install the python3-pip python3-dev

2. caffe 코드 clone하기

git clone https://github.com/BVLC/caffe

2. Makefile.config 설정하기

cd caffe
cp Makefile.confg.example Makefile.config
sudo vi Makefile.config

해당 파일에 아래의 값을 붙여넣기한다.

^[[200~CPU_ONLY := 1`
OPENCV_VERSION := 3
CUSTOM_CXX := g++
CUDA_DIR := /usr/local/cuda
CUDA_ARCH :=\
-gencode arch=compute_52,code=sm_52 \
-gencode arch=compute_60,code=sm_60 \
-gencode arch=compute_61,code=sm_61 \
-gencode arch=compute_61,code=compute_61
BLAS := atlas
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
PYTHON_LIB := /usr/lib
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib/x86_64-linux-gnu/hdf5/serial
BUILD_DIR := build
DISTRIBUTE_DIR := distribute
TEST_GPUID := 0
Q ?= @

본 글은, cuda 11.0 버전의 환경설정이다. 참고하도록.

a 입력 후, ctrl + v 하면 복붙된다. 참고로 복사할 때 모든 들여쓰기(space, enter)를 지우기를. 그러지 않으면 에러가 난다.

참고 블로그 : https://knight76.tistory.com/entry/makefile4-missing-separator-Stop-%ED%95%B4%EA%B2%B0%ED%95%98%EA%B8%B0

:makefile:4: *** missing separator. Stop. 해결하기

3. make pycaffe

sudo make all
sudo make test
sudo make runtest

마지막에 이런 화면이 나오면 성공

make pycaffe

만들어졌는지 여부 확인

echo $?

0이 출력되면 설치 완료

4. path 설정

export PYTHONPATH=/home/(사용자이름)/caffe/python:$PYTHONPATH

본인 컨테이너 이름을 작성하면 된다.

 

 

... 근데 python에서 import할 때 

Python 3.7.6 (default, Jan  8 2020, 19:59:22) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import caffe
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/2019112024/Desktop/caffe3/caffe/python/caffe/__init__.py", line 1, in <module>
    from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
  File "/home/2019112024/Desktop/caffe3/caffe/python/caffe/pycaffe.py", line 13, in <module>
    from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
ImportError: dynamic module does not define module export function (PyInit__caffe)

이런 에러가 남.