From 8c71168491deb41f0082a31d94bd36bc0967a1a8 Mon Sep 17 00:00:00 2001 From: Jiabin Yang Date: Tue, 25 Jun 2019 15:24:37 +0800 Subject: [PATCH] add nccl related info (#922) * add nccl related info * refine code --- doc/fluid/beginners_guide/install/install_CentOS.md | 10 +++++++++- doc/fluid/beginners_guide/install/install_Ubuntu.md | 8 ++++++++ 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/doc/fluid/beginners_guide/install/install_CentOS.md b/doc/fluid/beginners_guide/install/install_CentOS.md index 6a1a56369..e8c5a6384 100644 --- a/doc/fluid/beginners_guide/install/install_CentOS.md +++ b/doc/fluid/beginners_guide/install/install_CentOS.md @@ -22,7 +22,15 @@ CentOS系统下您可以使用`cat /proc/cpuinfo | grep avx`来检测您的处 * *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 + ## 选择如何安装 在CentOS的系统下我们提供4种安装方式： diff --git a/doc/fluid/beginners_guide/install/install_Ubuntu.md b/doc/fluid/beginners_guide/install/install_Ubuntu.md index 8dc394687..aa1f914b2 100644 --- a/doc/fluid/beginners_guide/install/install_Ubuntu.md +++ b/doc/fluid/beginners_guide/install/install_Ubuntu.md @@ -24,6 +24,14 @@ Ubuntu系统下您可以使用`cat /proc/cpuinfo | grep avx`来检测您的处 +* 请确保您已经正确安装nccl2，或者按照以下指令安装nccl2（这里提供的是ubuntu 16.04，CUDA9，cuDNN7下nccl2的安装指令），更多版本的安装信息请参考NVIDIA[官方网站](https://developer.nvidia.com/nccl/nccl-download): + + + wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb + dpkg -i nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb` + sudo apt-get install -y libnccl2=2.3.7-1+cuda9.0 libnccl-dev=2.3.7-1+cuda9.0 + + ## 选择如何安装 在Ubuntu的系统下我们提供4种安装方式： -- GitLab