Could not create cudnn handle: C

2021-04-19  本文已影响0人  热爱生活H_Li

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

上一篇下一篇

猜你喜欢

热点阅读