caffe配置NCCL

設置Makefile.configui

打開開關: USE_NCCL := 1, 並添加nccl庫路徑file

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

相關文章
相關標籤/搜索