Tensorflow GPU训练过程中遇到的问题总结

错误类型:CUDA_ERROE_OUT_OF_MEMORY GPU的全部memory资源不能全部都申请,可以通过修改参数来解决: 在session定义前增加 config = tf.ConfigProto(allow_soft_placement=True) #最多占gpu资源的70% gpu_options = tf.GPUOptions(per_process_gpu_memory_fract
相关文章
相关标签/搜索