torch.ones().cuda() vs. torch.ones_like() CPU利用率

BierOne

于 2023-08-18 20:28:52 发布

阅读量347

点赞数

CC 4.0 BY-SA版权

文章标签：深度学习人工智能 pytorch

本文链接：https://blog.youkuaiyun.com/BierOne/article/details/132368689

文章讨论了在PyTorch中使用`torch.ones()`和`torch.ones_like()`创建张量时，后者在将结果移动到GPU时能显著节省CPU资源，减少CPU占用率高达80%。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

    1. targets = torch.ones(outputs.shape).cuda() / num_classes
    2. targets = torch.ones_like(outputs) / num_classes

ones().cuda()会消耗大量的cpu （2000 CPU%），而直接ones_like能节省至 200