torch2.7.1与torch2.12.1在Windows gloo 后端的差异

72次阅读
没有评论

torch 2.7 的 Windows gloo 后端遇到 CUDA 张量时会静默走 CPU 回退路径:把张量拷到主机内存、做完规约、再拷回显存

torch 2.12 把这条回退路径拿掉了(或者说改成了直接对设备指针操作)。Windows 版 PyTorch 的 gloo 不带 GLOO_USE_CUDA,拿到显存地址当主机指针解引用

对于deepspeed需要world_size 守卫

正文完
 0
评论(没有评论)