GPU-communication-test

本文最后更新于:3 天前

背景

由于需要实现张量并行,考虑测试实验室153服务器的GPU通信指标。

GPU间需要大量的交换数据,于是GPU通信性能成为了非常重要的指标。NVIDIA推出的GPUDirect就是一组提升GPU通信性能的技术。但GPUDirect受限于PCI Expresss总线协议以及拓扑结构的一些限制,无法做到更高的带宽,为了解决这个问题,NVIDIA提出了NVLink总线协议。

测试方式

1
2
3
cd /usr/local/cuda/samples/1_Utilities/p2pBandwidthLatencyTest
make
./p2pBandwidthLatencyTest

即可得出GPU之间的通信速率


GPU-communication-test
http://example.com/2024/03/15/GPU-communication-test/
作者
dutsc
发布于
2024年3月15日
许可协议