作者:言_信者 | 来源:互联网 | 2023-07-11 18:13
篇首语:本文由编程笔记#小编为大家整理,主要介绍了ubuntu安装NCCL for cuda10.1相关的知识,希望对你有一定的参考价值。
配置:
nccl 下载链接:https://developer.nvidia.com/nccl/nccl-legacy-downloads
本人选择的是:NCCL 2.7.8, for CUDA 10.1, July 24,2020
下载本地文件后具体的操作链接是:
https://docs.nvidia.com/deeplearning/nccl/install-guide/index.html
本人总共使用的三条命令是:
sudo apt-key add /var/nccl-repo-2.7.8-ga-cuda10.1/7fa2af80.pub
sudo dpkg -i nccl-repo-ubuntu1804-2.7.8-ga-cuda10.1_1-1_amd64.deb
sudo apt update
sudo apt install libnccl2=2.7.8-1+cuda10.1 libnccl-dev=2.7.8-1+cuda10.1
因为是配合paddle框架使用,因此使用的检测成功的程序是:
import paddle
paddle.utils.run_check()
显示安装成功后的输出内容: