嘎嘎爽:使用NVIDIA_nccl和torch.nn.parallel.DistributedDataParallel一台服务器多显卡跑代码

本文介绍了在Ubuntu服务器上安装NVIDIACollectiveCommunicationsLibrary(NCCL)的步骤,包括检查软件需求(glibc和CUDA版本)、添加NCCL存储库、更新APT数据库、安装libnccl2和libnccl-dev包,以及如何在保留旧CUDA版本的情况下进行安装。
摘要由CSDN通过智能技术生成

承接上文:服务器联网后安装nccl

我参考官方安装教程:https://docs.nvidia.com/deeplearning/nccl/install-guide/index.html

我直接官方英译中

2.先决条件

2.1.软件要求
·glibc 2.17 或更高版本
·CUDA 10.0 或更高版本

3.安装NCCL

3.1.ubuntu
1.安装本地 NCCL 存储库
sudo dpkg -i nccl-repo-.deb
我的cuda driver是10.1所以选择安装包 nccl-repo-ubuntu1604-2.8.3-ga-cuda10.1_1-1_amd64.deb
即在终端输入命令:sudo dpkg -isudo dpkg -i nccl-repo-ubuntu1604-2.8.3-ga-cuda10.1_1-1_amd64.deb
2. 连网,更新 APT 数据库
sudo apt update
3. 连网,使用 APT 安装 libnccl2 软件包。 此外,如果您 需要使用 NCCL 编译应用程序,您也可以安装 libnccl-dev 包。保留旧版本的 CUDA。
sudo apt install libnccl2=2.8.3-1+cuda10.1 libnccl-dev=2.8.3-1+cuda10.1
我用的是sudo apt install libnccl2 libnccl-dev,感觉它自己会找依赖版本的安装包、、就这么简单!
在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值