安装nccl

1.安装nccl

git clone https://github.com/NVIDIA/nccl.git
cd nccl

打开makefile，修改路径

CUDA_HOME ?= /usr/local/cuda-9.0 (cuda路径)
PREFIX ?= /home/wangfeng/packages/nccl （安装路径）

make install

2.编译caffe
打开caffe CMakeLists.txt

caffe_option(USE_NCCL "Build Caffe with NCCL library support" ON)

打开caffe/cmake/Modules/FindNCCL.cmake

set(NCCL_INC_PATHS

    /home/wangfeng/packages/nccl/include

    )
set(NCCL_LIB_PATHS

     /home/wangfeng/packages/nccl/lib

     )

重新cmake caffe