我远程访问高性能计算节点，我不确定目录中是否安装了NVIDIA Collective Communications Library（NCCL），有没有办法检查是否安装了NCCL？
i2byvkas1#
你可以试试
locate nccl| grep "libnccl.so" | tail -n1 | sed -r 's/^.*\.so\.//'
或者如果使用PyTorch：
python -c "import torch;print(torch.cuda.nccl.version())"
检查此链接Command Cheatsheet: Checking Versions of Installed Software / Libraries / Tools for Deep Learning on Ubuntu对于容器，有时没有locate可用，可以用ldconfig -v替换它：
locate
ldconfig -v
ldconfig -v | grep "libnccl.so" | tail -n1 | sed -r 's/^.*\.so\.//'
9jyewag02#
通常可以在命令行中执行此操作：
nvcc --version
您可能需要运行：
sudo apt install nvidia-cuda-toolkit
我也是。正如另一位回答者提到的，您可以：
torch.cuda.nccl.version()
在pytorch中。复制粘贴到您的终端：
我确信在tensorflow 中有类似的东西。
2条答案
按热度按时间i2byvkas1#
你可以试试
或者如果使用PyTorch：
检查此链接Command Cheatsheet: Checking Versions of Installed Software / Libraries / Tools for Deep Learning on Ubuntu
对于容器，有时没有
locate可用，可以用ldconfig -v替换它：9jyewag02#
通常可以在命令行中执行此操作：
您可能需要运行：
我也是。
正如另一位回答者提到的，您可以：
在pytorch中。复制粘贴到您的终端：
我确信在tensorflow 中有类似的东西。