在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运行的时候提示错误:
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: