One Step Closer

Caffe docker 설치
GPU 버전의 Caffe와 같은 경우 nvidia-docker 플러그인이 추가적으로 설치 되어 있어야 함
$ docker pull bvlc/caffe:cpu // CPU only
$ docker pull bvlc/caffe:gpu // GPU only(cuDNN Caffe)

CUDA 설치
CUDA 및 cuDNN 라이브러리 관련 docker 이미지 및 docker shell 실행
$ nvidia-docker pull nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04
$ nvidia-docker run -it -v /data:/data nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 /bin/bash --name myname

Caffe 빌드에 필요한 패키지 설치(docker 내 sudo 권한 필요없음)
첫 실행시 업데이트 정보 최신화
$ apt-get update;apt-get upgrade
$ apt-get install build-essential cmake libblas-dev libboost-all-dev libprotobuf-dev libgoogle-glog-dev libgflags-dev libhdf5-dev libhdf5-serial-dev libopencv-dev liblmdb-dev libleveldb-dev libsnappy-dev protobuf-compiler libatlas-dev libatlas-base-dev git

Caffe 소스코드 다운로드
$ git clone https://github.com/bvlc/caffe /data/caffe
$ cd /data/caffe && git checkout tags/1.0
$ cp Makefile.config.example Makefile.config

Makefile에서 cuDNN 설정

# cuDNN acceleration switch (uncomment to build with cuDNN).

USE_CUDNN := 1

caffe 빌드 및 설치
$ mkdir build
$ cd build
$ cmake ..
$ make all
$ make install