看 warning 信息可知,是缺少 nccl
在英偉達官網(wǎng),根據(jù)自己的 cuda 選擇對應的nccl,這里以cuda11.2為例,后面根據(jù)自己的 nccl2?和 cuda 版本修改相關命令
也可使用命令行下載
下載完成
安裝鏡像庫
安裝 nccl
將nccl添加到環(huán)境變量中