could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
在CPU下运行没有问题,但是在GPU版本下,一直运行不成功。可以尝试限制一下GPU的运行资源,添加如下代码:config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))sess = tf.Session(config=config)刚开始分配少量资源,然后按需慢慢增加GPU资源。...
原创
2021-01-10 20:30:28 ·
287 阅读 ·
1 评论