Docker로 Caffe 설치(GPU Mode)

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

2. 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

3. 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

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

5. Makefile에서 cuDNN 설정

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

   USE_CUDNN := 1

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