Open hrwangcn opened 1 year ago
需要自行配置tensorflow config
import tensorflow as tf physical_devices = tf.config.list_physical_devices('GPU') tf.config.experimental.set_memory_growth(physical_devices[0], True)
需要自行配置tensorflow config