美文网首页
Could not create cudnn handle: C

Could not create cudnn handle: C

作者: 热爱生活H_Li | 来源:发表于2021-04-19 08:28 被阅读0次

cudnn已经读进去了


image.png

加如下代码即可

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
image.png

引用:https://blog.csdn.net/pkuyjxu/article/details/89402298

相关文章

网友评论

      本文标题:Could not create cudnn handle: C

      本文链接:https://www.haomeiwen.com/subject/emlulltx.html