安装NCCL 2.1.15
安装环境:ubuntu16.04, cuda9.1版本, cudnn7.1.3
因为新版本的tensorflow依赖nccl2.x以上版本,github仓库只有1.x版本。需要在官网下载nccl2版本。
参考步骤—下载nccl-repo-ubuntu1604-2.1.15-ga-cuda9.1_1-1_amd64.deb
- 官网下载地址
- 注册,选择对应的ubuntu版本和 cuda版本
- 1.Install the repository.
sudo dpkg -i nccl-repo-<version>.deb
- 2.Update the APT database: