caffe配置NCCL

设置Makefile.configui

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

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

相关文章
相关标签/搜索