Ubuntu16.04安装NCCL

官方NCCL安装说明https://docs.nvidia.com/deeplearning/sdk/nccl-install-guide/index.html

1、下载NCCL

https://developer.nvidia.com/nccl
在这里插入图片描述

2、安装存储库

对于本地存储库

sudo dpkg -i nccl-repo-<version>.deb

对于网络存储库

sudo dpkg -i nvidia-machine-learning-repo-<version>.deb

具体方式跟据下载的NCCL存储库类型选择,这里下的本地版。
所以是:

sudo dpkg -i nccl-repo-ubuntu1604-2.2.13-ga-cuda9.0_1-1_amd64.deb

3、更新APT数据库

sudo apt update

4、使用APT安装libnccl2软件包。 此外,如果需要使用NCCL编译应用程序,还可以安装libnccl-dev软件包

注意:如果使用的是网络存储库,则以下命令会将CUDA升级到最新版本。

sudo apt install libnccl2 libnccl-dev

如果希望保留较旧版本的CUDA,请指定特定版本,例如:

sudo apt-get install libnccl2=2.2.13-1+cuda9.0 libnccl-dev=2.2.13-1+cuda9.0

没报错,安装成功
在这里插入图片描述

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值