Ubuntu 安装 NCCL 2.0

https://developer.nvidia.com/nccl
sudo dpkg -i nccl-repo-ubuntu1604-2.2.13-ga-cuda9.0_1-1_amd64.deb
sudo apt update
sudo apt install libnccl2 libnccl-dev
sudo cp /usr/lib/x86_64-linux-gnu/libnccl.so.2 /usr/local/cuda/lib
sudo cp /usr/include/nccl.h /usr/local/cuda/include/

猜你喜欢

转载自blog.csdn.net/u013431916/article/details/82759525