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()

显示安装成功后的输出内容:
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值