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

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐