1. 程式人生 > >關於執行tensorflow出現錯誤解決方案

關於執行tensorflow出現錯誤解決方案

1.Process finished with exit code -1073740791 (0xC0000409)

2.could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

3.could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM

4Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms) 

出現上面錯誤有可能是視訊記憶體問題,解決方案:

更改下面如下引數

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.1)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

如根據上面分配給tensorflow的GPU視訊記憶體大小為:GPU實際視訊記憶體*0.1

可以按照需要,設定不同的值,來分配視訊記憶體,我的per_process_gpu_memory_fraction=1時,執行錯誤,所以我改為0.1就好了