可以透過以下方法讓限制記憶體的使用。 import tensorflow as tf # 設定自動增長GPU 記憶體用量 gpu_options = tf.GPUOptions(allow_growth=True)
確定! 回上一頁