Tensorflow -Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

我在做卷积层可视话的时候碰到的这个问题,其实不是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代码的后面,要不然依旧会报错

你可能感兴趣的:(python,tensorflow)