import tensorflow as tf
gpu_device_name = tf.test.gpu_device_name()
print(gpu_device_name)
gpu_options = tf.GPUOptions()
gpu_options.allow_growth = True
config_photo = tf.ConfigProto(allow_so
python tensorflow-gpu的session中config参数设置
最新推荐文章于 2024-04-24 13:35:29 发布