python-3.x - Keras、Tensorflow 在模型构建时保留所有 GPU 内存

Question

我的 GPU 是 NVIDIA RTX 2080 TI

Keras 2.2.4

TensorFlow-GPU 1.12.0

CUDA 10.0

一旦我加载构建模型（在编译之前），我发现 GPU 内存已完全分配

[0] GeForce RTX 2080 钛 | 50'C, 15 % | 10759 / 10989 MB | issd/8067(10749M)

可能是什么原因，我该如何调试它？

即使我通过生成器加载，我也没有多余的内存来加载数据

我试图监控 GPU 内存使用情况，发现它在构建层之后（在编译模型之前）已满

标签： python-3.xtensorflowkeras

Answer 1

当我加载预训练的 ResNet50 时，我遇到了类似的问题。GPU 内存使用量激增至 11GB，而 ResNet50 通常仅消耗不到 150MB。

就我而言，问题是我还导入了 PyTorch 而没有在我的代码中实际使用它。评论后，一切正常。但我有另一台 PC 具有相同的代码，工作得很好。所以我卸载并重新安装了正确版本的 Tensorflow 和 PyTorch。即使我导入 PyTorch，一切正常。