Tensorflow 报错 CUBLAS_STATUS_ALLOC_FAILED
·
windows环境下,tf不会分配所有可用的内存,因此需要手动设置,允许动态内存分配增长。
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
reference:
https://stackoverflow.com/questions/41117740/tensorflow-crashes-with-cublas-status-alloc-failed
更多推荐
已为社区贡献11条内容
所有评论(0)