Ubuntu20.04安装nccl
时间: 2023-10-26 14:08:20 浏览: 201
详解ubuntu 20.04 LTS安装记录
安装 NVIDIA Collective Communications Library(NCCL)可以提高多 GPU 系统的性能。以下是在 Ubuntu 20.04 上安装 NCCL 的步骤:
1. 添加 NVIDIA 的 APT 仓库。可以通过以下命令完成:
```
$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.0-470.57.02-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.0-470.57.02-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
$ sudo apt-get update
```
2. 安装 NCCL。可以通过以下命令完成:
```
$ sudo apt-get install libnccl2 libnccl-dev
```
安装完成后,可以使用 NCCL 库来优化多 GPU 系统中的通信。
阅读全文