需要下载以及安装的包:
https://developer.nvidia.com/nccl/nccl-legacy-downloads 根据cuda版本下载,我下载的是:
nccl-local-repo-ubuntu2004-2.8.4-cuda11.1_1.0-1_amd64.deb 使用以下命令安装:
sudo dpkg -i nccl-local-repo-ubuntu2004-2.8.4-cuda11.1_1.0-1_amd64.deb 安装完,按照提示把pub安装一下
sudo apt-key add /var/nccl-local-repo-ubuntu2004-2.8.4-cuda11.1/7fa2af80.pub 然后执行以下命令:
sudo apt update
sudo apt install libnccl2