caffe NCCL

[기록용 포스트] 제가 참고하기 위한 포스트입니다. 원작자 분 감사합니다.

출처 : http://jangjy.tistory.com/265 

[Caffe] Digits을 위한 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