1. 程式人生 > >Ubuntu 安裝 NCCL 2.0

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/