Ubuntu 18.04安装NCCL
1. 下载
官网下载地址:https://developer.nvidia.com/nccl/nccl-download
注意版本与你的cuda适配。
2.安装
获得.deb安装文件如:nccl-local-repo-ubuntu1804-2.8.4-cuda11.1_1.0-1_amd64.deb后
sudo dpkg -i nccl-local-repo-ubuntu1804-2.8.4-cuda11.1_1.0-1_amd64.deb # 安装
# 如果提示缺少公共CUDA GPG秘钥
sudo apt-key add /var/nccl-repo-2.8.3-ga-cuda10.2/7fa2af80.pub
# 必不可少更新
sudo apt update
sudo apt install libnccl2 libnccl-dev