From b73f0b7d3013b9df4bece7dabd114806595441dc Mon Sep 17 00:00:00 2001 From: xsrobin <50069408+xsrobin@users.noreply.github.com> Date: Tue, 25 Jun 2019 16:09:31 +0800 Subject: [PATCH] add nccl restrict (#925) --- doc/fluid/beginners_guide/install/install_CentOS.md | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/doc/fluid/beginners_guide/install/install_CentOS.md b/doc/fluid/beginners_guide/install/install_CentOS.md index ef8e3bda6..90230b980 100644 --- a/doc/fluid/beginners_guide/install/install_CentOS.md +++ b/doc/fluid/beginners_guide/install/install_CentOS.md @@ -20,6 +20,15 @@ CentOS系统下您可以使用`cat /proc/cpuinfo | grep avx2`来检测您的处 * *CUDA 工具包9.0配合cuDNN v7* * *CUDA 工具包8.0配合cuDNN v7* * *GPU运算能力超过1.0的硬件设备* + +* 请确保您已经正确安装nccl2,或者按照以下指令安装nccl2(这里提供的是CentOS 7,CUDA9,cuDNN7下nccl2的安装指令),更多版本的安装信息请参考NVIDIA[官方网站](https://developer.nvidia.com/nccl/nccl-download): + + + wget http://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/x86_64/nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm + rpm -i nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm + sudo apt-get install -y libnccl2=2.3.7-1+cuda9.0 libnccl-dev=2.3.7-1+cuda9.0 + yum update -y + yum install -y libnccl-2.3.7-2+cuda9.0 libnccl-devel-2.3.7-2+cuda9.0 libnccl-static-2.3.7-2+cuda9.0 -- GitLab