Kaggle GPU 显示内存错误,而在 Colab 上运行良好(相同的 GPU 后端)

问题描述 投票:0回答:0

我正在研究一个笔记本,它比较同一模型的 Keras 和 Pytorch 实现。该笔记本在 Colab 中运行良好,但在提交给 Kaggle 时,它正在抛出

CUDA error: out of memory
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.` 

一旦它击中 PyTorch 模型。

为什么它在 Colab 上工作而不在 Kaggle 上工作的任何想法,尽管两者都有 15GB 的 VRAM 可用。 Colab 几乎没有达到 2.5 GB VRAM,但 Kaggle 正在将 VRAM 最大化到 15 GB。

tensorflow pytorch gpu tf.keras kaggle
© www.soinside.com 2019 - 2024. All rights reserved.