ubuntu安装NCCL for cuda10.1

Posted MarToony|名角

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了ubuntu安装NCCL for cuda10.1相关的知识,希望对你有一定的参考价值。

配置:

  • 多块GPU
  • 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()

显示安装成功后的输出内容:

以上是关于ubuntu安装NCCL for cuda10.1的主要内容,如果未能解决你的问题,请参考以下文章

Ubuntu18.04安装Cuda10.1

ubuntu16.04 安装cuda 10.1 和cudnn

Ubuntu 16.04 上安装 CUDA 10.1和cuDNN v8.0.5

ubuntu20.04系统+cuda10.1+opencv4.1.0+dense_flow的编译与安装,吐血总结!!!

如何使用 Tensorflow V.2.4 RTX 2070 Super Ubuntu 18.04 安装 Cuda 10.1

Anaconda3+CUDA10.1+CUDNN7.6+TensorFlow2.6安装(Ubuntu16)