Ubuntu安装NCCL

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
Ubuntu安装NCCL_第1张图片

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信息
Ubuntu安装NCCL_第2张图片

你可能感兴趣的:(Linux/Ubuntu,机器学习/深度学习,nccl)