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