synchronize

西西弗Sisyphus3 小时前
pytorch·python·synchronize
PyTorch 中用于 主机（CPU）与设备（GPU）同步 的函数 torch.cuda.synchronize()flyfish完整代码在文末GPU 是 异步执行 的CPU 发送指令给 GPU（比如矩阵乘法、卷积） CPU 不会等 GPU 算完，直接跑去执行下一行代码 GPU 在后台默默计算，计算完了再通知 CPU
我是有底线的