03 2022 档案

关于pytorch分布式训练dist.barrier()产生死锁的问题
摘要：1. 安装nccl库支持 2. 导入环境变量： vim /etc/profile export NCCL_P2P_DISABLE=1 export NCCL_IB_DISABLE=1https://gitee.com/573363031/baidu_ai_security_advbox/blob/m 阅读全文
posted @ 2022-03-26 19:22 风影旋新月 阅读(1860) 评论(0) 推荐(0) 编辑
vscode调试分布式代码
摘要：命令行： CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node 1 --master_port 12345 main.py --eval --cfg configs/swin_base_patch4_wi 阅读全文
posted @ 2022-03-23 14:46 风影旋新月 阅读(674) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示