krrmjssf
仙凡之谣
2018-11-23 03:18

用tensorflow跑卷积网络报错:InternalError (see above for traceback): cudnn PoolForward launch failed

  • tensorflow
  • 神经网络
  • gpu
  • internalerror

图片说明
用tf跑一个卷积神经网络时报错!
我查看了很多关于报该类型的错误文档,大多都说是因为tensorflow是一个很贪心的框架,总是试图为自己分配全部的GPU资源,但实际上GPU资源不可能完全一点不被其他程序占用,导致初始化失败。解决的办法都是再创建session时加上配置参数:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = Session(config=tf.ConfigProto(gpu_options=gpu_options))
还有许多其他的参数,我都试了,但还是报了相同的错误,心里苦,跪求各位dalao!

  • 点赞
  • 收藏
  • 复制链接分享

2条回答

为你推荐

换一换