一种人工智能训练集群的拥塞控制方法、芯片及介质

本发明公开了一种人工智能训练集群的拥塞控制方法、芯片及介质，涉及人工智能技术领域。该方法，包括：解析数据接收端反馈的确认数据包；在确认数据包中携带的基准速率未变化，且当前基准速率调整间隔不小于传输速率调整间隔阈值时，根据单向队列延时以及目标排队时间，对数据包发送速率进行分阶段调整，得到目标发送速率；根据目标发送速率以及环回时间确定拥塞窗口数量，并根据拥塞窗口数量确定数据包发送间隔。本发明实施例的技术方案能够对人工智能训练集群进行精准的拥塞控制，可保证训练节点高效可靠的通信，并兼具良好的可扩展性。