cuda版本11.2
说明PaddlePaddle需要安装NCCL
登录官网
找到对应版本
官方提供了多种安装方式,本文使用Local installers (x86)本地安装
点击对应的版本下载如:
nccl-local-repo-ubuntu1804-2.8.4-cuda11.2_1.0-1_amd64.deb
安装步骤:
sudo dpkg -i nccl-local-repo-ubuntu1804-2.8.4-cuda11.2_1.0-1_amd64.deb
# 更新源镜像
sudo apt update
# 安装 nccl
sudo apt install libnccl2=2.8.4-1+cuda11.2 libnccl-dev=2.8.4-1+cuda11.2
安装nccl的命令来源于: