python - 如何检查NCCL的版本

Question

我正在远程访问高性能计算节点。我不确定 NVIDIA 集体通信库 (NCCL) 是否安装在我的目录中？有没有办法检查NCCL

Answer 1

你可以试试locate nccl| grep "libnccl.so" | tail -n1 | sed -r 's/^.*\.so\.//'或 torch.cuda.nccl.version()如果你使用 PyTorch
检查此链接 Command Cheatsheet: Checking Versions of Installed Software / Libraries / Tools for Deep Learning on Ubuntu
对于容器，没有 locate有时可用，可以将其替换为 ldconfig -v :ldconfig -v | grep "libnccl.so" | tail -n1 | sed -r 's/^.*\.so\.//'