ubuntu 安装 nccl(最全)

1、多GPUs进行并行计算,Caffe自带实现. 在多个 GPU 上运行 Caffe 需要使用 NVIDIA NCCL。

刚开始装的时候,我也踩了许多坑,直到在谷歌上找到NVIDIA的官方文档,才有效的解决了安装的问题。

官方nccl的安装说明:https://docs.nvidia.com/deeplearning/sdk/nccl-install-guide/index.html

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

我主要是参考安装文档中的下面三句命令:

因为在官网上我下的是网络存储库版本,(ps:但是我一直以为是本地存储库,请大家一定要看清自己下的安装包的名称,对号入座

之后,在我的服务器上输入:


sudo dpkg -i nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb


sudo apt update


sudo apt install libnccl2=2.2.13-1+cuda9.0 libnccl-dev=2.2.13-1+cuda9.0

没有报错,安装成功。

更多精彩请关注公众号【AI算法与数学之美】

或者扫下方图片二维码奥

别忘记关注公众号与点  “在看

 

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值