1.下载NCCL
官网下载地址
根据想要安装的nccl版本以及自己cuda版本选择对应下载项,以我自己安装的情况为例,选择下载:Download NCCL v2.3.7, for CUDA 10.0, Nov 8 & Dec 14, 2018
nccl下载安装分本地安装(Local installers (x86))和网络安装(Network installers (x86)), 这里我们选择网络安装,我的系统时Ubuntu16.04,所以选择:Network Installer for Ubuntu16.04
。下载完成后,得到一个文件:nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
2.安装
安装网络版本依次执行以下命令即可:
sudo dpkg -i nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
sudo apt update
sudo apt install libnccl2=2.3.7-1+cuda10.0 libnccl-dev=2.3.7-1+cuda10.0
成功安装大概会出现下面类似log信息