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
相關文章
相關標籤/搜索