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