NCCL (NVIDIA Colloctive Comunications Library)是英伟达的一款直接与GPU交互的库。
安装cupy前需要先安装该库。
1. 下载
官网下载地址:https://developer.nvidia.com/nccl/nccl-download
注意版本与你的cuda适配。
2.安装
获得.deb安装文件如:nccl-repo-ubuntu1804-2.8.3-ga-cuda10.2_1-1_amd64.deb后
sudo dpkg -i nccl-repo-ubuntu1804-2.8.3-ga-cuda10.2_1-1_amd64.deb # 安装
# 如果提示缺少公共CUDA GPG秘钥
sudo apt-key add /</

本文介绍了如何在Ubuntu 18.04上安装NCCL库,该库是NVIDIA用于GPU间通信的集合通信库。首先,需要从官方地址下载与CUDA版本匹配的NCCL,然后安装.deb包来完成安装。
最低0.47元/天 解锁文章
2611

被折叠的 条评论
为什么被折叠?



