TensorFlow GPU OOM Error

說明

遇到這個報錯是因為GPU 的顯存爆掉了，不夠大，所以程式崩潰，這邊有幾個方法第一個換硬體顯存擴大，不然就是在程式運行的時候減少圖片的大小。

還有第三個方法擴大顯存，增加下方的程式碼。

操作流程

code

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

如果不填入2，空白的話代表不使用GPU

REF

• GPU顯存

• 解決OMM錯誤