Windows版Tensorflow显存不足解决方案 (2017-07-23 00:24:27)

在Windows版Tensorflow运行的时候提示错误:

failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

在官方Github上找到一个解决方案

创建session的时候允许显存增长

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

with tf.Session(config = config) as sess:

你可能感兴趣的:(Windows版Tensorflow显存不足解决方案 (2017-07-23 00:24:27))