-
import tensorflow as tf
-
import numpy as np
-
import keras
-
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
-
config.gpu_options.per_process_gpu_memory_fraction = 0.3
-
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
显存不足
最新推荐文章于 2024-02-22 18:55:46 发布