我在做卷积层可视话的时候碰到的这个问题,其实不是cuDNN配置的问题,而是GPU占用,可通过下面的代码解决。
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
值得注意的是,
import tensorflow as tf
应在上述限制GPU代码的后面,要不然依旧会报错