1.安装nccl
&& git clone https://github.com/NVIDIA/nccl.git \
&& pip3 install cython -i https://pypi.tuna.tsinghua.edu.cn/simple \
&& cd nccl && make -j32 src.build CUDA_HOME='/usr/local/cuda-9.0/' NVCC_GENCODE="-gencode=arch=compute_61,code=sm_61" \
&& yum install -y rpm-build rpmdevtools && make -j32 pkg.redhat.build && make install \
&& cd /home && rm -rf nccl \
# 验证 nccl 是否安装成功
# cd /home && git clone https://github.com/NVIDIA/nccl-tests.git \
# cd nccl-tests && make -j32 && ./build/all_reduce_perf -b 8 -e 256M -f 2 -g 2 \
# cd /home && rm -rf nccl-tests
# cat /etc/ld.so.conf
include ld.so.conf.d/*.conf
# echo "/usr/local/lib" >> /etc/ld.so.conf
# ldconfig
2.安装git