[Caffe] Digits을 위한 NCCL 설정 방법

Multi-GPU 사용을 위해 사용되는 NCCL.

가끔 재설치 할때 삽질하는 경우가 있어서 기록.

NCCL은 NVIDIA github 에서 받을 수 있다 (https://github.com/NVIDIA/nccl)

(master 버전이 아닌 cuda-8.0 대응 버전을 받으라는 얘기도 있어서 얘를 받음)

1. 압축 해제

2. 압축을 푼 폴더에서 

$ sudo make install

3. build 폴더안의 폴더들을 /usr/local/nccl 폴더로 복사 (include, lib, obj)

$ sudo cp -r /nccl경로/build /usr/local/nccl

4. 경로설정 ( 본인은 ~/.bashrc 파일에 설정)

$ sudo vim ~/.bashrc

$ # nccl

$ export NCCL_ROOT_DIR=/usr/local/nccl

$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/nccl/lib

여기까지 하면 NCCL 을 이용하여 multi-gpu 사용이 가능하다.