张量流中的CUDA内存分配失败

Question

我正在尝试解决tensorflow中的CUDA内存分配失败问题，请帮帮我。 使用： cuda = 10 tensorflow == 1.14

  

E tensorflow / stream_executor / cuda / cuda_driver.cc：828]失败   从设备分配5.40G（5798205696字节）：   CUDA_ERROR_OUT_OF_MEMORY：内存不足E   tensorflow / stream_executor / cuda / cuda_driver.cc：828]分配失败   来自设备的4.86G（5218384896字节）：CUDA_ERROR_OUT_OF_MEMORY：内存不足E tensorflow / stream_executor / cuda / cuda_driver.cc：828]无法   从设备分配4.37G（4696546304字节）：   CUDA_ERROR_OUT_OF_MEMORY：内存不足LLVM错误：内存不足

我尝试了一些已经存在但无法使用的soln。 我尝试过：

1。

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.9)
  config = tf.ConfigProto(gpu_options=gpu_options)

2。

config = tf.configConfigProto()
  config.gpu_options.allow_growth = True/False

我正在开发一个项目，该项目已经在github上了。 如果有人需要更多详细信息，请检查repo。