failed to allocate 5.91G (6347372032 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory 

import tensorflow as tf
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

 

试试

posted @ 2021-04-09 14:57  bH1pJ  阅读(436)  评论(0)  编辑  收藏  举报