風格轉移：OOM分配時，張

Question

我一直在試圖運行這個Tensorflow風格轉移實現 - https://github.com/anishathalye/neural-style在Windows（的GPU版本），但我得到這個錯誤：

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[64,239400] [[Node: gradients/MatMul_grad/MatMul = MatMul[T=DT_FLOAT, transpose_a=false, transpose_b=true, _device="/job:localhost/replica:0/task:0/gpu:0"](gradients/truediv_2_grad/tuple/control_dependency, Reshape)]]

我是在Tensorflow和Python中都是完整的初學者，所以我不知道如何解決這個問題。

Answer 1

這是內存不足錯誤。您沒有足夠的GPU內存來運行此映像的深層網絡。

你有2個解決方案：

1. 如果你不關心速度，使用CPU的版本，因爲你可能比GPU的內存更多的CPU內存（RAM）。您將CUDA_VISIBLE_DEVICES設置爲禁用GPU：CUDA_VISIBLE_DEVICES= python neural_style.py <content file> --styles <style file> --output <output file>
2. 處理較小的圖像。提供更小尺寸的圖像可以真正減少內存使用量。因此，請在將圖像提供給腳本之前嘗試對圖像進行降採樣;您可以嘗試多種尺寸來查看可以放入GPU內存的最大尺寸。