대상 : 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 |