我正在研究一个笔记本,它比较同一模型的 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。