TensorFlow GPU OOM Error

markdown #說明 遇到這個報錯是因為GPU 的顯存爆掉了，不夠大，所以程式崩潰，這邊有幾個方法第一個換硬體顯存擴大，不然就是在程式運行的時候減少圖片的大小。 還有第三個方法擴大顯存，增加下方的程式碼。 #操作流程 ##code ``` import os os.environ["CUDA_VISIBLE_DEVICES"] = "2" ``` 如果不填入2，空白的話代表不使用GPU ##REF - GPU顯存 - 解決OMM錯誤