NCCL学习笔记-安装验证

安装与验证

https://developer.nvidia.com/nccl
安装后将 NCCL 的库路径添加到 LD_LIBRARY_PATH(Linux)中。
安装验证

python -c "import torch; print(torch.distributed.is_nccl_available())"

如果输出为 True,则说明 NCCL 已成功安装并与 PyTorch 集成。

### 安装 NCCL 使用 pkg-config 的指南 为了使用 `pkg-config` 来管理 NCCL 库的编译选项,需要先确保系统上已经正确安装NCCL 和 `pkg-config` 工具。 #### 准备工作 确认已安装必要的依赖项和工具链。对于 Ubuntu 系统来说,可以通过以下命令来更新包列表并安装基础开发环境: ```bash sudo apt-y build-essential devscripts debhelper dh-make dkms fakeroot git libnuma-dev python3-all python3-setuptools wget zip unzip zlib1g-dev ``` #### 下载与安装 NCCL 获取适用于当前 CUDA 版本的 NCCL 软件包。假设正在使用的操作系统版本为 Ubuntu 18.04 并搭配 CUDA 10.0[^3],可以按照 NVIDIA 提供的方法下载官方发布的二进制文件或源码构建最新稳定版。 如果选择通过 APT 存储库方式简化安装过程,则可执行如下操作: ```bash wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo apt-get install nvidia-cuda-toolkit libcudnn7=7.6.5.32-1+cuda10.0 nccl-repo-ubuntu1804-2.4.7-ga-cuda10.0 ``` 上述命令会自动处理 NCCL 及其依赖关系的安装,并将其注册到系统的 `pkg-config` 数据库中。 #### 验证安装成功与否 完成以上步骤之后,应该能够利用 `pkg-config` 查询 NCCL 编译标志以及链接器参数: ```bash nccl_include=$(pkg-config --cflags-only-I nccl | sed 's/-I//') echo "Include path for NCCL headers is ${nccl_include}" nccl_lib=$(pkg-config --libs-only-L nccl | sed 's/-L//') echo "Library search paths are ${nccl_lib}" ``` 此外还可以尝试编写简单的测试程序验证 NCCL 是否正常运作。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

MatsumotoChrikk

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值