NCCL (NVIDIA Colloctive Comunications Library)是英伟达的一款直接与GPU交互的库。
安装cupy前需要先安装该库。
1. 下载
官网下载地址:https://developer.nvidia.com/nccl/nccl-download
注意版本与你的cuda适配。
2.安装
获得.deb安装文件如:nccl-repo-ubuntu1804-2.8.3-ga-cuda10.2_1-1_amd64.deb后
sudo dpkg -i nccl-repo-ubuntu1804-2.8.3-ga-cuda10.2_1-1_amd64.deb # 安装
# 如果提示缺少公共CUDA GPG秘钥
sudo apt-key add