대상 : https://github.com/weiliu89/caffe/tree/ssd


1. ssd 파일 다운로드

git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd



2. caffe 설치

# Modify Makefile.config according to your Caffe installation. cp Makefile.config.example Makefile.config make -j8


# Make sure to include $CAFFE_ROOT/python to your PYTHONPATH.

gedit ~/.bashrc # CAFFE_ROOT와 PYTHONPATH를 추가

# 터미널 재시작


conda create -n ssd python=2

source activate ssd

make py

pip install -r python/requirements.txt


python

import caffe # 별 반응없이 넘어가면 성공 !!




2-1. make 시 오류


google/protobuf/arena.h: No such file or directory


https://github.com/google/protobuf/blob/master/src/README.md

에 가서 protobuf 설치


아나콘다 사용 시

conda uninstall libprotobuf

conda uninstall protobuf



hdf5.h: No such file or directory


sudo apt-get install libhdf5-10
sudo apt-get install libhdf5-serial-dev
sudo apt-get install libhdf5-dev
sudo apt-get install libhdf5-cpp-11
find /usr -iname "*hdf5.h*"
/usr/include/hdf5/serial/hdf5.h
export CPATH="/usr/include/hdf5/serial/"


출처 : https://github.com/NVIDIA/DIGITS/issues/156


2-2. import 시 오류


dynamic module does not define module export function


아나콘다로 python 2.7 가상환경 만들어서

python/requirements.txt 설치로 해결함



3. 학습 시 오류


error == cudaSuccess (10 vs. 0)  invalid device ordinal


examples/ssd/ssd_pascal.py

332번째 줄에서

gpus = "0,1,2,3"  --> gpus = "0"

으로 수정

'실습 > 설치' 카테고리의 다른 글

Ubuntu 16.04 LTS 설치 및 딥러닝 환경 설치  (0) 2018.01.12
python requirements.txt 설치하기  (0) 2017.12.28
opencv 설치  (0) 2017.08.05
gcc-5, g++-5 설치  (0) 2017.07.02
opencv 설치  (0) 2017.07.02

+ Recent posts