caffe配置NCCL

设置Makefile.config

打开开关: USE_NCCL := 1, 并添加nccl库路径

USE_NCCL := 1
INCLUDE_DIRS += /path/nccl/build/include
LIBRARY_DIRS += /path/nccl/build/lib

posted @ 2017-07-15 16:59  BlueOceans  阅读(2866)  评论(0编辑  收藏  举报