TensorFlow 按需使用显存设置

2019-10-20  本文已影响0人  酷酷滴小爽哥

方式如下：

gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
上一篇 下一篇

猜你喜欢

热点阅读