- 博客(1)
- 收藏
- 关注
原创 could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
在跑深度学习代码时,我们可能会遇到以下一种情况,源错误代码如下: 以上问题在GPU上运行代码,可能是显卡不够用,但在CPU上运行是正常的,我们知道CPU和GPU运行代码的速度相差是非常大的,这时我们只需在源代码上添加两行代码就可以解决这个问题,意思是在一开始占用显卡的内存不要太高,然后按需求慢慢增加内存。添加的代码如下:config = tf.ConfigProto(g...
2019-11-04 21:33:31
292
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人