gpt4 book ai didi

deep-learning - 在 RTX2070 上加载 ResNet50 - 内存不足

转载 作者:行者123 更新时间:2023-12-04 03:53:25 24 4
gpt4 key购买 nike

<分区>

我正在尝试加载 ResNext50,在它的 CenterNet 之上,我可以使用 Google Colab 或 Kaggle 的 GPU 来完成。但是,

  1. 想知道这个网络需要多少 GPU 内存 (VRAM)?

  2. 当使用 RTX 2070 时剩余 5.5GB VRAM(超出 8GB),我无法加载它。

批量大小为 1,#of workers 为 1,一切都设置为最小值。操作系统:Ubuntu 18.04(使用 PyTorch)

在 TensorFlow 中,我知道我可以限制 VRAM 的数量(这使我能够加载和运行网络,尽管我没有足够的 VRAM),但在 PyTorch 中我还没有找到这个功能。

有什么解决办法吗?

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com