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

但是,我的訓練batch_size是比測試batch_size大的,也就是說,訓練的時候比測試是更容易出現不夠的情況的。是以覺得奇怪。
查了别人寫的文章,試了一些方法:
- 隔離GPU,在代碼中添加 os.environ[“CUDA_VISIBLE_DEVICES”] = “-1”
- 指定GPU,添加 os.environ[“CUDA_VISIBLE_DEVICES”] = “2”
這兩個方法都可以起作用。但是有些方法不行,比如指定另一塊GPU,或者限制GPU使用率。我沒有繼續去探究是為什麼,若有知道原因的人,還請不吝賜教,謝謝!