练习到第六章,发现1060的6G不够了

打印:2020-01-09 22:36:08.018462: E tensorflow/stream_executor/cuda/cuda_driver.cc:806] failed to allocate 5.93G (6370295808 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory

图像太吃内存,换到其它章节看看。

One thought on “练习到第六章,发现1060的6G不够了”

  1. 有解,可以在python中设在使用部分内存:增加arg参数 –gpu_memory_fraction 0.8,表示使用80%内存,实验ok,可以继续第六章。

Leave a Reply

Your email address will not be published. Required fields are marked *