Admin` 2023-04-26 21:42 采纳率: 0%
浏览 6

torch.matmul()

在使用torch.matmul时遇到cuda out of memory如何解决,共四维,最后两个维度分别为65536 8,8 65536

  • 写回答

1条回答 默认 最新

  • youcans_ 人工智能领域优质创作者 2023-04-26 23:09
    关注

    内存爆炸,还能有什么方法呢?
    (1)扩大显存
    (2)减小 batch

    评论

报告相同问题?

问题事件

  • 创建了问题 4月26日