【Caffe】nccl的安装

本文详细介绍了如何从源代码编译安装NVIDIA的NCCL库,并将其集成到Caffe深度学习框架中,以加速多GPU训练过程。通过设置makefile.config文件中的USE_NCCL参数,可以启用Caffe对NCCL的支持。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

$ git clone https://github.com/NVIDIA/nccl.git
$ cd nccl
$ sudo make install -j8
$ sudo ldconfig

 出现以下情况表示编译成功。由于是编译到了默认的路径,所以caffe的makefile.config中不需要修改成INCLUDE_DIRS += /path/nccl/build/include    LIBRARY_DIRS += /path/nccl/build/lib

 

 然后直接打开makefile.config中的nccl接口

USE_NCCL := 1

然后编译caffe

make all -j $(($(nproc) + 1))  
make test -j $(($(nproc) + 1))  
make runtest -j $(($(nproc) + 1))  
make pycaffe -j $(($(nproc) + 1))

参考

https://docs.nvidia.com/deeplearning/sdk/nccl-install-guide/index.html

https://github.com/NVIDIA/nccl

https://github.com/NVIDIA/nccl/releases

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值