天天看點

keras報錯:Process finished with exit code -1073740791 (0xC0000409)

訓練3D-Unet的時候,訓練過程沒有問題,但是在預測的時候,報了這個錯誤,上網查,說是顯存不夠。

keras報錯:Process finished with exit code -1073740791 (0xC0000409)

但是,我的訓練batch_size是比測試batch_size大的,也就是說,訓練的時候比測試是更容易出現不夠的情況的。是以覺得奇怪。

查了别人寫的文章,試了一些方法:

  1. 隔離GPU,在代碼中添加 os.environ[“CUDA_VISIBLE_DEVICES”] = “-1”
  2. 指定GPU,添加 os.environ[“CUDA_VISIBLE_DEVICES”] = “2”

這兩個方法都可以起作用。但是有些方法不行,比如指定另一塊GPU,或者限制GPU使用率。我沒有繼續去探究是為什麼,若有知道原因的人,還請不吝賜教,謝謝!

繼續閱讀